Update app.py

app.py

@@ -97,14 +97,14 @@ def prod_function(transformer_model, prod_dl, webcam_dl):
     for batch in acclerated_webcam_dl:
         with torch.no_grad():
             #img_prod = acclerated_prod_data['pixel_values']
-            emb_prod = accelerated_model(batch)
+            emb_prod = accelerated_model(batch['pixel_values'])
 
     prod_preds = []
 
     for batch in acclerated_prod_dl:
         #img = batch['pixel_values']
         with torch.no_grad():
-            emb = accelerated_model(batch)
+            emb = accelerated_model(batch['pixel_values'])
         distance = F.pairwise_distance(emb, emb_prod)
 
         prod_preds.append(distance)