Update app.py

app.py

@@ -4,7 +4,7 @@ from huggingface_hub import hf_hub_download
 
 
 def download_models(model_id):
-    hf_hub_download("SakshiRathi77/void-space-detection/weights", filename=f"{model_id}", local_dir=f"./")
+    hf_hub_download("SakshiRathi77/void-space-detection", filename=f"{model_id}", local_dir=f"./")
     return f"./{model_id}"
 
 @spaces.GPU
@@ -48,9 +48,9 @@ def app():
                 model_path = gr.Dropdown(
                     label="Model",
                     choices=[
-                        "best.pt"
+                        "state_dict.pt"
                     ],
-                    value="best.pt",
+                    value="state_dict.pt",
                 )
                 image_size = gr.Slider(
                     label="Image Size",