nick5363 commited on
Commit
3e7dc6c
·
verified ·
1 Parent(s): 0d815b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -17
app.py CHANGED
@@ -1,22 +1,12 @@
1
  import gradio as gr
2
- from ultralytics import YOLO
3
  from PIL import Image
4
- import numpy as np
5
 
6
- # Load model từ file
7
- model = YOLO("model.pt")
 
8
 
9
- def detect_pattern(img):
10
- results = model(img)
11
- annotated_img = results[0].plot() # Vẽ kết quả lên ảnh
12
- return Image.fromarray(annotated_img)
13
 
14
- demo = gr.Interface(
15
- fn=detect_pattern,
16
- inputs=gr.Image(type="pil", label="Upload Chart"),
17
- outputs=gr.Image(type="pil", label="Detected Pattern"),
18
- title="Stock Pattern Detection (YOLOv8)",
19
- description="Upload a stock chart image to detect patterns."
20
- )
21
-
22
- demo.launch()
 
1
  import gradio as gr
 
2
  from PIL import Image
 
3
 
4
+ # Load hình ảnh từ local
5
+ def show_pattern():
6
+ return Image.open("pattern.webp") # đổi tên đúng với file bạn upload
7
 
8
+ with gr.Blocks(title="Chart Patterns") as app:
9
+ gr.Markdown("## Popular Chart Patterns")
10
+ gr.Image(show_pattern, elem_id="pattern-image", show_label=False)
 
11
 
12
+ app.launch()