Update src/streamlit_app.py

src/streamlit_app.py

@@ -21,11 +21,8 @@ def get_model():
     device = "cuda:0" if torch.cuda.is_available() else "cpu"
     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)
+    model, device = load_model("/app/src/model.pt", device=device)
+    
     return model, device
 
 model, device = get_model()