Update src/streamlit_app.py

src/streamlit_app.py

@@ -19,7 +19,13 @@ uploaded_file = st.file_uploader("Upload CSV or FASTA", type=["csv", "fasta"])
 @st.cache_resource
 def get_model():
     device = "cuda:0" if torch.cuda.is_available() else "cpu"
-    model, device = load_model("model.pt", device=device)
+    from huggingface_hub import hf_hub_download
+
+    model_path = hf_hub_download(
+        repo_id="caokai1073/StriMap",
+        filename="src/model.pt",
+    )
+    model, device = load_model(model_path, device=device)
     return model, device
 
 model, device = get_model()