Update app.py

app.py

@@ -8,7 +8,7 @@ from huggingface_hub import hf_hub_download
 # Helper function to download models from Hugging Face
 def get_model_path(model_name):
     model_cache_path = hf_hub_download(
-        repo_id="atalaydenknalbant/budgerigar_yolo_models", 
+        repo_id="atalaydenknalbant/budgerigar_models", 
         filename=model_name
     )
     return model_cache_path
@@ -39,7 +39,7 @@ interface = gr.Interface(
     inputs=[
         gr.Image(type="pil", label="Upload Image"),
         gr.Radio(
-            choices=['budgerigar_yolo11x.pt', 'budgerigar_yolov9e.pt', 'budgerigar_yolo11s.pt', 'budgerigar_yolo11n.pt'],
+            choices=['budgerigar_yolo11x.pt', 'budgerigar_yolov9e.pt', 'budgerigar_yolo11s.pt', 'budgerigar_yolo11n.pt', 'rtdetr-x.pt'],
             label="Model Name",
             value="budgerigar_yolo11x.pt",
         ),
@@ -50,7 +50,7 @@ interface = gr.Interface(
     outputs=gr.Image(type="pil", label="Annotated Image"),
     cache_examples=True,
     title="Budgerigar Gender Determination",
-    description="Pretrained YOLO models for determining budgerigar gender based on cere color variations. Upload image(s) for inference",
+    description="Pretrained Object detection models for determining budgerigar gender based on cere color variations. Upload image(s) for inference",
     examples=[
         ["Male.png", "budgerigar_yolov9e.pt", 0.25, 0.45, 300],
         ["Female.png", "budgerigar_yolo11x.pt", 0.25, 0.45, 300],