Update app.py

app.py

@@ -1,16 +1,18 @@
 import gradio as gr
+import io
 from ultralytics import YOLO
 import cv2
 import numpy as np
 from PIL import Image
 import json
-from fastapi import FastAPI
 
 # Load your custom YOLO model
-model = YOLO("fentanyl_oft.pt")
+model = YOLO("0829_kh_11_fenh_ft_544.pt")
 
 def detect_keypoints(image):
-    # Your existing detection code...
+    """
+    Run YOLO inference and return keypoints data
+    """
     try:
         # Convert PIL Image to numpy array
         if isinstance(image, Image.Image):
@@ -59,26 +61,16 @@ def detect_keypoints(image):
     except Exception as e:
         return {"success": False, "error": str(e)}
 
-# Create FastAPI app for API endpoints
-app = FastAPI()
-
-@app.post("/api/detect")
-async def api_detect_keypoints(file: bytes):
-    try:
-        # Convert bytes to PIL Image
-        image = Image.open(io.BytesIO(file))
-        result = detect_keypoints(image)
-        return result
-    except Exception as e:
-        return {"success": False, "error": str(e)}
-
-# Create Gradio interface
+# Create Gradio interface with API access enabled
 iface = gr.Interface(
     fn=detect_keypoints,
     inputs=gr.Image(type="pil"),
     outputs=gr.JSON(),
-    title="YOLO Keypoint Detection"
+    title="YOLO Keypoint Detection",
+    description="Upload an image to detect keypoints using custom YOLO model",
+    api_name="predict"  # This enables API access at /api/predict
 )
 
-# Mount Gradio on FastAPI
-app = gr.mount_gradio_app(app, iface, path="/")
+# Launch with API enabled
+if __name__ == "__main__":
+    iface.launch(share=False)