Update app.py

app.py

@@ -239,9 +239,8 @@ def prod_function(model_pretrained, prod_ds):
     # Evaluate at the end of the epoch
     accelerated_model.eval()
 
-    videos = acclerated_prod_ds['pixel_values']
     with torch.no_grad():
-        outputs = accelerated_model(videos)
+        outputs = accelerated_model(acclerated_prod_ds)
 
     prod_logits = outputs.squeeze(1)
     prod_pred = prod_logits.argmax(-1)