mamechin
commited on
Commit
•
f5b910d
1
Parent(s):
caf7e51
v12
Browse files- app.py +2 -2
- requirements.txt +1 -1
app.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import gradio as gr
|
2 |
-
import
|
3 |
# import torch
|
4 |
-
model =
|
5 |
|
6 |
def predict(input_image):
|
7 |
"""
|
|
|
1 |
import gradio as gr
|
2 |
+
from ultralytics import YOLO
|
3 |
# import torch
|
4 |
+
model = YOLO('best.pt')
|
5 |
|
6 |
def predict(input_image):
|
7 |
"""
|
requirements.txt
CHANGED
@@ -1,2 +1,2 @@
|
|
1 |
torch
|
2 |
-
|
|
|
1 |
torch
|
2 |
+
ultralytics
|