fcernafukuzaki commited on
Commit
2b07e2c
1 Parent(s): 8b10c9a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
- #modelo_yolo = YOLO('best.pt')
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'''