Update app.py

app.py

@@ -11,9 +11,9 @@ import gradio as gr
 
 DEVICE = config.device
 
-MODEL = BERTBaseUncased()
-MODEL.load_state_dict(torch.load(config.MODEL_PATH, map_location=torch.device(DEVICE)))
-MODEL.eval()
+# MODEL = BERTBaseUncased()
+# MODEL.load_state_dict(torch.load(config.MODEL_PATH, map_location=torch.device(DEVICE)))
+# MODEL.eval()
 
 
 
@@ -58,10 +58,10 @@ def sentence_prediction(sentence):
 
     # device = config.device
 
-    # model = BERTBaseUncased()
+    model = BERTBaseUncased()
     # model.load_state_dict(torch.load(
     #     model_path, map_location=torch.device(device)))
-    # model.to(device)
+    model.to(device)
 
     outputs, [] = engine.predict_fn(test_data_loader, MODEL, device)
     print(outputs)