UI Fixes

app.py

@@ -134,7 +134,7 @@ def detect_objects(image):
 
     return annotated_image, detection_summary
 
-# Gradio Interface
+# Enhanced Gradio Interface with Links to Model Card and Repository
 iface = gr.Interface(
     fn=detect_objects,
     inputs=gr.Image(type="pil"),
@@ -143,12 +143,14 @@ iface = gr.Interface(
         gr.Textbox(label="Detections")
     ],
     title="JunkWaxHero ⚾ - Baseball Card Set Detection (ONNX Model)",
-    description="Upload an image to itentify the set of the baseball card (1980-1999).",
+    description=(
+        "Upload an image to identify the set of the baseball card (1980-1999).\n\n"
+        "[🔗 Model Card & Repository](https://huggingface.co/enusbaum/JunkWaxHero)"
+    ),
     examples=["examples/card1.jpg", "examples/card2.jpg", "examples/card3.jpg"],
     theme="default",  # You can choose other themes if desired
     allow_flagging="never"  # Disable flagging if not needed
-    # Removed 'layout' parameter
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()