Update app.py

app.py

@@ -287,11 +287,11 @@ with app_tab:
             prediction = prod_function(app, image_paths, picture)
             #predictions = torch.cat(prediction, 0).to(device)
             #match_idx = torch.argmin(predictions)
-            st.write(predictions)
+            st.write(prediction)
             st.write(image_paths)
         
             # Display the results
-            if predictions[match_idx] >= 0.9:
+            if prediction[match_idx] >= 0.9:
               st.write('Welcome: ',image_paths[match_idx].split('/')[-1].split('.')[0])
             else:
               st.write("Match not found")