Update app.py

app.py

@@ -1,28 +1,22 @@
-
 import gradio as gr
 from ultralytics import YOLO
-import cv2
+from PIL import Image
 import numpy as np
 
-model = YOLO("foduucom/stockmarket-pattern-detection-yolov8")
+# Load model từ file
+model = YOLO("model.pt")
 
-def detect_pattern(image):
-    results = model(image)
-    if results and results[0].plot:
-        return cv2.cvtColor(results[0].plot(), cv2.COLOR_RGB2BGR)
-    return image
+def detect_pattern(img):
+    results = model(img)
+    annotated_img = results[0].plot()  # Vẽ kết quả lên ảnh
+    return Image.fromarray(annotated_img)
 
-with gr.Blocks(theme=gr.themes.Base()) as app:
-    gr.Markdown("# Stock Market Pattern Detection (YOLOv8)")
-    with gr.Tabs():
-        with gr.Tab("Upload Image"):
-            image_input = gr.Image(type="numpy")
-            image_output = gr.Image()
-            detect_btn = gr.Button("Detect Pattern")
-            detect_btn.click(fn=detect_pattern, inputs=image_input, outputs=image_output)
-        with gr.Tab("Real-time (Camera)"):
-            webcam_input = gr.Image(source="webcam", streaming=True)
-            webcam_output = gr.Image()
-            webcam_input.change(fn=detect_pattern, inputs=webcam_input, outputs=webcam_output)
+demo = gr.Interface(
+    fn=detect_pattern,
+    inputs=gr.Image(type="pil", label="Upload Chart"),
+    outputs=gr.Image(type="pil", label="Detected Pattern"),
+    title="Stock Pattern Detection (YOLOv8)",
+    description="Upload a stock chart image to detect patterns."
+)
 
-app.launch()
+demo.launch()