Update app.py

app.py

@@ -285,7 +285,7 @@ with app_tab:
         with st.spinner("Wait for it...", show_time=True):
             # Run the predictions
             prediction = prod_function(app, image_paths, picture)
-            #predictions = torch.cat(prediction, 0).to(device)
+            prediction = torch.cat(prediction, 0).to(device)
             match_idx = torch.argmax(prediction)
             st.write(prediction)
             st.write(image_paths)