Spaces:
Runtime error
Runtime error
abrar-adnan
commited on
Commit
•
9a80d11
1
Parent(s):
09a17ed
updated app to use v2
Browse files
app.py
CHANGED
@@ -41,7 +41,7 @@ vehicle_labels = (
|
|
41 |
'Yacht'
|
42 |
)
|
43 |
|
44 |
-
model = load_learner('vehicle-recognizer-
|
45 |
|
46 |
def recognize_image(image):
|
47 |
pred, idx, probs = model.predict(image)
|
|
|
41 |
'Yacht'
|
42 |
)
|
43 |
|
44 |
+
model = load_learner('vehicle-recognizer-v2.pkl')
|
45 |
|
46 |
def recognize_image(image):
|
47 |
pred, idx, probs = model.predict(image)
|