Update app.py

app.py

@@ -9,6 +9,7 @@ import numpy as np
 import pandas as pd
 import torch
 import os
+from tqdm import tqdm
 
 # [theme]
 # base="dark"
@@ -162,7 +163,7 @@ if st.button('확인'):
     predictions , true_labels = [], []
 
     # Predict 
-    for batch in range(len(test_dataloader)):
+    for batch in tqdm(test_dataloader):
         # Add batch to GPU
         batch = tuple(t.to(device) for t in batch)