pcreem commited on
Commit
e2bd5a8
·
1 Parent(s): b42e7a0
Files changed (4) hide show
  1. README.md +4 -4
  2. app.py +12 -0
  3. best0628.pt +3 -0
  4. requirements.txt +3 -0
README.md CHANGED
@@ -1,10 +1,10 @@
1
  ---
2
  title: Mulas
3
- emoji:
4
- colorFrom: purple
5
- colorTo: blue
6
  sdk: gradio
7
- sdk_version: 5.35.0
8
  app_file: app.py
9
  pinned: false
10
  ---
 
1
  ---
2
  title: Mulas
3
+ emoji: 🦀
4
+ colorFrom: red
5
+ colorTo: pink
6
  sdk: gradio
7
+ sdk_version: 5.34.2
8
  app_file: app.py
9
  pinned: false
10
  ---
app.py ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from ultralytics import YOLO
3
+ import gradio as gr
4
+
5
+ model = YOLO("best0628.pt") # 或者是從 hf_hub_download 取得
6
+
7
+ def detect(image):
8
+ results = model(image)
9
+ return results[0].plot() # 回傳帶框圖像
10
+
11
+ iface = gr.Interface(fn=detect, inputs="image", outputs="image")
12
+ iface.launch(show_api=True)
best0628.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1536806a99f859860f48938f253c99edd6da12af5de188fb5ecd50ea58f89a90
3
+ size 22516707
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ ultralytics
2
+ gradio
3
+ torch