Spaces:
Build error
Build error
fcernafukuzaki
commited on
Commit
•
2b07e2c
1
Parent(s):
8b10c9a
Update app.py
Browse files
app.py
CHANGED
@@ -26,7 +26,7 @@ import gradio as gr
|
|
26 |
import torch
|
27 |
from PIL import Image
|
28 |
from ultralytics import YOLO
|
29 |
-
from ultralyticsplus import render_result
|
30 |
|
31 |
# Images
|
32 |
torch.hub.download_url_to_file('https://i.pinimg.com/originals/7f/5e/96/7f5e9657c08aae4bcd8bc8b0dcff720e.jpg', 'ejemplo1.jpg')
|
@@ -45,7 +45,7 @@ class YOLODetect():
|
|
45 |
return self.results
|
46 |
|
47 |
def show(self):
|
48 |
-
render = render_result(model=self.modelo, image=self.source, result=self.results[0])
|
49 |
return render
|
50 |
|
51 |
def to_json(self):
|
@@ -66,7 +66,7 @@ class YOLODetect():
|
|
66 |
def to_dataframe(self):
|
67 |
return pd.DataFrame(self.json_result)
|
68 |
|
69 |
-
|
70 |
|
71 |
def yolo(size, iou, conf, im):
|
72 |
'''Wrapper fn for gradio'''
|
|
|
26 |
import torch
|
27 |
from PIL import Image
|
28 |
from ultralytics import YOLO
|
29 |
+
#from ultralyticsplus import render_result
|
30 |
|
31 |
# Images
|
32 |
torch.hub.download_url_to_file('https://i.pinimg.com/originals/7f/5e/96/7f5e9657c08aae4bcd8bc8b0dcff720e.jpg', 'ejemplo1.jpg')
|
|
|
45 |
return self.results
|
46 |
|
47 |
def show(self):
|
48 |
+
render = None #render_result(model=self.modelo, image=self.source, result=self.results[0])
|
49 |
return render
|
50 |
|
51 |
def to_json(self):
|
|
|
66 |
def to_dataframe(self):
|
67 |
return pd.DataFrame(self.json_result)
|
68 |
|
69 |
+
modelo_yolo = YOLO('best.pt')
|
70 |
|
71 |
def yolo(size, iou, conf, im):
|
72 |
'''Wrapper fn for gradio'''
|