Update app.py

app.py

@@ -53,8 +53,7 @@ def predict(im):
     probabilities = torch.nn.functional.softmax(out[0], dim=0)
     values, indices = torch.topk(probabilities, 5)
     
-    query = values[0]
-    print(query, values)
+    query = LABELS[indices[0]]
 
     n_results=3
     text_embeddings = compute_text_embeddings([query]).detach().numpy()