Update app.py

app.py

@@ -4,7 +4,7 @@ from huggingface_hub import hf_hub_download
 
 
 def download_models(model_id):
-    hf_hub_download("merve/yolov9", filename=f"{model_id}", local_dir=f"./")
+    hf_hub_download("SakshiRathi77/void-space-detection/weights", filename=f"{model_id}", local_dir=f"./")
     return f"./{model_id}"
 
 @spaces.GPU
@@ -48,12 +48,9 @@ def app():
                 model_path = gr.Dropdown(
                     label="Model",
                     choices=[
-                        "gelan-c.pt",
-                        "gelan-e.pt",
-                        "yolov9-c.pt",
-                        "yolov9-e.pt",
+                        "best.pt"
                     ],
-                    value="gelan-e.pt",
+                    value="best.pt",
                 )
                 image_size = gr.Slider(
                     label="Image Size",
@@ -93,34 +90,34 @@ def app():
             outputs=[output_numpy],
         )
         
-        gr.Examples(
-            examples=[
-                [
-                    "data/zidane.jpg",
-                    "gelan-e.pt",
-                    640,
-                    0.4,
-                    0.5,
-                ],
-                [
-                    "data/huggingface.jpg",
-                    "yolov9-c.pt",
-                    640,
-                    0.4,
-                    0.5,
-                ],
-            ],
-            fn=yolov9_inference,
-            inputs=[
-                img_path,
-                model_path,
-                image_size,
-                conf_threshold,
-                iou_threshold,
-            ],
-            outputs=[output_numpy],
-            cache_examples=True,
-        )
+        # gr.Examples(
+        #     examples=[
+        #         [
+        #             "data/zidane.jpg",
+        #             "gelan-e.pt",
+        #             640,
+        #             0.4,
+        #             0.5,
+        #         ],
+        #         [
+        #             "data/huggingface.jpg",
+        #             "yolov9-c.pt",
+        #             640,
+        #             0.4,
+        #             0.5,
+        #         ],
+        #     ],
+        #     fn=yolov9_inference,
+        #     inputs=[
+        #         img_path,
+        #         model_path,
+        #         image_size,
+        #         conf_threshold,
+        #         iou_threshold,
+        #     ],
+        #     outputs=[output_numpy],
+        #     cache_examples=True,
+        # )
 
 
 gradio_app = gr.Blocks()
@@ -135,7 +132,6 @@ with gradio_app:
         """
         <h3 style='text-align: center'>
         Follow me for more!
-        <a href='https://twitter.com/kadirnar_ai' target='_blank'>Twitter</a> | <a href='https://github.com/kadirnar' target='_blank'>Github</a> | <a href='https://www.linkedin.com/in/kadir-nar/' target='_blank'>Linkedin</a>  | <a href='https://www.huggingface.co/kadirnar/' target='_blank'>HuggingFace</a>
         </h3>
         """)
     with gr.Row():