princeml commited on
Commit
6d13a4e
·
1 Parent(s): dc62026

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import numpy as np
4
  from glob import glob
5
  from models import Yolov4
6
  import gradio as gr
7
- model = Yolov4(weight_path="yolov4.weights", class_name_path='coco_classes.txt')
8
  def gradio_wrapper(img):
9
  global model
10
  #print(np.shape(img))
 
4
  from glob import glob
5
  from models import Yolov4
6
  import gradio as gr
7
+ model = Yolov4(weight_path="best.pt", class_name_path='coco_classes.txt')
8
  def gradio_wrapper(img):
9
  global model
10
  #print(np.shape(img))