Update app.py

app.py

@@ -128,7 +128,7 @@ if st.button('predict'):
         preds = []
         model.eval()
         model.to(device)
-        tk0 = enumerate(test_loader, total=len(test_loader))
+        tk0 = enumerate(test_loader)
         for i, inputs in tk0:
             for k, v in inputs.items():
                 inputs[k] = v.to(device)