Update src/streamlit_app.py

src/streamlit_app.py

@@ -61,10 +61,10 @@ def load_caption_model():
     ckpt_name = 'aehrc/medicap'
     
     local_folder = "model2/"
-    if os.path.exists(local_folder):
-        medicap = transformers.AutoModel.from_pretrained(local_folder, trust_remote_code=True)
-    else:
-        medicap = transformers.AutoModel.from_pretrained(ckpt_name, trust_remote_code=True)
+    # if os.path.exists(local_folder):
+    medicap = transformers.AutoModel.from_pretrained(local_folder, trust_remote_code=True)
+    # else:
+        # medicap = transformers.AutoModel.from_pretrained(ckpt_name, trust_remote_code=True)
     medicap = medicap.to(device)
     medicap.eval()