Update app.py

app.py

@@ -24,6 +24,10 @@ cap_labels = cap_labels = {
     'visor cap'
 }
 
+version = 1
+model_path = f"cap-recognizer-v{version}.pkl"
+model = load_learner(model_path)
+
 def recognize_image(image):
     pred, idx, probs = model.predict(image)
     return dict(zip(sorted(cap_labels), map(float, probs)))