Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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='
|
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)
|