Update app.py

app.py

@@ -9,9 +9,13 @@ import config
 
 import gradio as gr
 
-MODEL = None
 DEVICE = config.device
 
+MODEL = BERTBaseUncased()
+MODEL.load_state_dict(torch.load(config.MODEL_PATH, map_location=torch.device(DEVICE)))
+MODEL.eval()
+
+
 
 T = tokenizer.TweetTokenizer(
     preserve_handles=True, preserve_hashes=True, preserve_case=False, preserve_url=False)
@@ -59,15 +63,12 @@ def sentence_prediction(sentence):
     #     model_path, map_location=torch.device(device)))
     # model.to(device)
 
-    outputs, [] = engine.predict_fn(test_data_loader, model, device)
+    outputs, [] = engine.predict_fn(test_data_loader, MODEL, device)
     print(outputs)
     return {"label":outputs[0]}
 
 if __name__ == "__main__":
-    MODEL = BERTBaseUncased()
-    MODEL.load_state_dict(torch.load(
-        config.MODEL_PATH, map_location=torch.device(DEVICE)))
-    MODEL.eval()
+   
     
     demo = gr.Interface(
         fn=sentence_prediction,