Update app.py

app.py

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