Hugging Face's logo

Spaces:
abrar-adnan
/
vehicle-recognizer
Running

App Files Community
abrar-adnan commited on
Commit
9a80d11
•
1 Parent(s): 09a17ed

updated app to use v2

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ vehicle_labels = (
41
  'Yacht'
42
  )
43
 
44
- model = load_learner('vehicle-recognizer-v1.pkl')
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)