nick5363 commited on
Commit
4c4fc7d
·
verified ·
1 Parent(s): 85b8292

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +28 -0
  2. requirements.txt +4 -0
app.py ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ import gradio as gr
3
+ from ultralytics import YOLO
4
+ import cv2
5
+ import numpy as np
6
+
7
+ model = YOLO("foduucom/stockmarket-pattern-detection-yolov8")
8
+
9
+ def detect_pattern(image):
10
+ results = model(image)
11
+ if results and results[0].plot:
12
+ return cv2.cvtColor(results[0].plot(), cv2.COLOR_RGB2BGR)
13
+ return image
14
+
15
+ with gr.Blocks(theme=gr.themes.Base()) as app:
16
+ gr.Markdown("# Stock Market Pattern Detection (YOLOv8)")
17
+ with gr.Tabs():
18
+ with gr.Tab("Upload Image"):
19
+ image_input = gr.Image(type="numpy")
20
+ image_output = gr.Image()
21
+ detect_btn = gr.Button("Detect Pattern")
22
+ detect_btn.click(fn=detect_pattern, inputs=image_input, outputs=image_output)
23
+ with gr.Tab("Real-time (Camera)"):
24
+ webcam_input = gr.Image(source="webcam", streaming=True)
25
+ webcam_output = gr.Image()
26
+ webcam_input.change(fn=detect_pattern, inputs=webcam_input, outputs=webcam_output)
27
+
28
+ app.launch()
requirements.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+
2
+ ultralytics
3
+ opencv-python-headless
4
+ gradio