mamechin commited on
Commit
f5b910d
1 Parent(s): caf7e51
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
- import yolov7
3
  # import torch
4
- model = model = yolov7.load('best.pt', device="cpu", hf_model=True, trace=False)
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
- yolov7detect
 
1
  torch
2
+ ultralytics