Update app.py

app.py

@@ -36,19 +36,19 @@ def md_loading():
 #    model.load_state_dict(ckpt['model_state_dict'])
     
 #    device = torch.device("cuda" if torch.cuda.is_available() and not False else "cpu")
-    device = torch.device("cpu")
+#    device = torch.device("cpu")
         
-    model.to(device)
+#    model.to(device)
     
     label_tbl = np.load('./label_table.npy')
     loc_tbl = pd.read_csv('./kisc_table.csv', encoding='utf-8')
 
     print('ready')
 
-    return tokenizer, model, label_tbl, loc_tbl, device
+    return tokenizer, model, label_tbl, loc_tbl
 
 # 모델 로드
-tokenizer, model, label_tbl, loc_tbl, device = md_loading()
+tokenizer, model, label_tbl, loc_tbl = md_loading()
 
 
 # 데이터 셋 준비용
@@ -162,7 +162,7 @@ if st.button('확인'):
     # Predict 
     for batch in range(test_dataloader):
         # Add batch to GPU
-        batch = tuple(t.to(device) for t in batch)
+#        batch = tuple(t.to(device) for t in batch)
 
         # Unpack the inputs from our dataloader
         test_input_ids, test_attention_mask = batch