Upload app.py with huggingface_hub

app.py

@@ -1,39 +1,33 @@
 """Chart Pattern Detection API — YOLOv8"""
 import gradio as gr
 import json
-import traceback
+from ultralyticsplus import YOLO
 
-from ultralytics import YOLO
-from huggingface_hub import hf_hub_download
-
-model_path = hf_hub_download(repo_id="foduucom/stockmarket-pattern-detection-yolov8", filename="model.pt")
-model = YOLO(model_path)
+model = YOLO("foduucom/stockmarket-pattern-detection-yolov8")
+model.overrides["conf"] = 0.20
+model.overrides["iou"] = 0.45
 print(f"Model loaded. Classes: {model.names}")
 
 def detect_patterns(image):
     try:
         if image is None:
             return json.dumps({"patterns": [], "error": "No image"})
-        results = model.predict(source=image, conf=0.20, iou=0.45, imgsz=640, verbose=False)
+        results = model.predict(source=image, imgsz=640)
         patterns = []
         for r in results:
             if r.boxes is None or len(r.boxes) == 0:
                 continue
             for i in range(len(r.boxes)):
                 box = r.boxes[i]
-                cls_id = int(box.cls[0])
-                conf = float(box.conf[0])
-                xyxy = box.xyxy[0].tolist()
-                label = r.names.get(cls_id, f"class_{cls_id}")
                 patterns.append({
-                    "label": label,
-                    "confidence": round(conf, 3),
-                    "bbox": [round(x, 1) for x in xyxy],
+                    "label": r.names.get(int(box.cls[0]), "unknown"),
+                    "confidence": round(float(box.conf[0]), 3),
+                    "bbox": [round(float(x), 1) for x in box.xyxy[0].tolist()],
                 })
         patterns.sort(key=lambda p: p["confidence"], reverse=True)
         return json.dumps({"patterns": patterns, "count": len(patterns)})
     except Exception as e:
-        return json.dumps({"patterns": [], "error": str(e), "trace": traceback.format_exc()})
+        return json.dumps({"patterns": [], "error": str(e)})
 
 demo = gr.Interface(
     fn=detect_patterns,