Update app.py

app.py

@@ -213,6 +213,8 @@ if picture is not None:
     prediction = prod_function(model_pretrained, prod_dl, webcam_dl)
     predictions = torch.cat(prediction, 0).to(device)
     match_idx = torch.argmin(predictions)
+    st.write(predictions)
+    st.write(image_paths)
 
     # Display the results
     if predictions[match_idx] <= 0.3: