update app.py

app.py

@@ -71,6 +71,7 @@ def chat_completion(
     top_k: int,
     top_p: float,
 ):
+    global model
     if model is None:
         reload_model(DEFAULT_MODEL)
     prompt = generate_alpaca_prompt(instruction=message)
@@ -91,8 +92,7 @@ def chat_completion(
 
 def reload_model(model_name: str):
     global model
-    if 'model' in globals():
-        del model
+    model = None
     gc.collect()
     model = Llama.from_pretrained(
         **model_paths[model_name],
@@ -107,7 +107,7 @@ def reload_model(model_name: str):
         height=500,
         placeholder='<strong>Hi doctor, I have a headache, what should I do?</strong>',
         label=model_name,
-        avatar_images=[None, 'https://raw.githubusercontent.com/minhnguyent546/medical-llama2/demo/assets/med_alpaca.png'],  # pyright: ignore[reportArgumentType]
+        avatar_images=[None, 'https://raw.githubusercontent.com/minhnguyent546/medical-llama2/refs/heads/master/assets/medical_llama2.png'],  # pyright: ignore[reportArgumentType]
     )
     return app_title_mark, chatbot
 
@@ -139,7 +139,7 @@ def main() -> None:
                     height=500,
                     placeholder='<strong>Hi doctor, I have a headache, what should I do?</strong>',
                     label=DEFAULT_MODEL,
-                    avatar_images=[None, 'https://raw.githubusercontent.com/minhnguyent546/medical-llama2/demo/assets/med_alpaca.png'],  # pyright: ignore[reportArgumentType]
+                    avatar_images=[None, 'https://raw.githubusercontent.com/minhnguyent546/medical-llama2/refs/heads/master/assets/medical_llama2.png'],  # pyright: ignore[reportArgumentType]
                 )
                 textbox = gr.Textbox(
                     placeholder='Hi doctor, I have a headache, what should I do?',