Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ import os
|
|
4 |
|
5 |
def getResult(filename):
|
6 |
model_name='best.pt'
|
7 |
-
model = torch.hub.load( "",'custom', source='local', path = model_name, force_reload = True)
|
8 |
img = "/content/"+filename+""
|
9 |
results = model(img)
|
10 |
return results
|
|
|
4 |
|
5 |
def getResult(filename):
|
6 |
model_name='best.pt'
|
7 |
+
model = torch.hub.load( " ",'custom', source='local', path = model_name, force_reload = True)
|
8 |
img = "/content/"+filename+""
|
9 |
results = model(img)
|
10 |
return results
|