Update app.py

app.py

@@ -6,6 +6,8 @@ import json
 import torch
 import gradio as gr
 
+from config import model_config
+
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 model_dir = snapshot_download(model_config['model_dir'])