Update app.py

app.py

@@ -6,6 +6,7 @@ import torch
 from accelerate import Accelerator
 import accelerate
 import time
+import gradio as gr
 
 model = None
 tokenizer = None
@@ -38,7 +39,7 @@ def load_model(model_name = "zl111/ChatDoctor", eight_bit=0, device_map="auto"):
 
     generator = model.generate
 
-load_model("./pretrained")
+load_model()
 
 history = []
 
@@ -93,3 +94,4 @@ def go():
 
 while True:
     go()
+