Update app.py

app.py

@@ -162,7 +162,7 @@ if st.button('확인'):
     predictions , true_labels = [], []
 
     # Predict 
-    for batch in range(test_dataloader):
+    for batch in range(len(test_dataloader)):
         # Add batch to GPU
 #        batch = tuple(t.to(device) for t in batch)