Update app.py

app.py

@@ -24,7 +24,12 @@ def predict_image(image):
     )
 
     pred = classifier.predict(img=image)
-    return pred
+
+    label2id = json.load(open("./models/" + colname + "/best_model.pth"))["label2id"].keys()
+    vec = [100.0 if a.lower() == pred.lower() else 0.00 for a in label2id]
+    
+    return vec
+    # return pred
 
 # open categories.txt in read mode
 categories = open("categories.txt", "r")