sundea commited on
Commit
50999b8
·
1 Parent(s): a20948f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -204,7 +204,7 @@ class options:
204
  def get_output(image):
205
  image.save(f"{BASE_DIR}/input/image.jpg")
206
  source = f"{BASE_DIR}/input"
207
- opt = options(weights='logo_detection.pt',source=source)
208
  bbox = None
209
  with torch.no_grad():
210
  # if opt.update: # update all models (to fix SourceChangeWarning)
 
204
  def get_output(image):
205
  image.save(f"{BASE_DIR}/input/image.jpg")
206
  source = f"{BASE_DIR}/input"
207
+ opt = options(weights='best.pt',source=source)
208
  bbox = None
209
  with torch.no_grad():
210
  # if opt.update: # update all models (to fix SourceChangeWarning)