Update app.py

app.py

@@ -11,6 +11,12 @@ import gradio as gr
 
 # DEVICE = config.device
 
+import requests
+URL = "https://huggingface.co/FFZG-cleopatra/bert-emoji-latvian-twitter/blob/main/pytorch_model.bin"
+response = requests.get(URL)
+open("pytorch_model.bin", "wb").write(response.content)
+
+
 model = AutoModel.from_pretrained("thak123/bert-emoji-latvian-twitter-classifier")
 # 7 EPOCH Version
 BERT_PATH = "FFZG-cleopatra/bert-emoji-latvian-twitter"
@@ -72,12 +78,12 @@ def sentence_prediction(sentence):
 
     # device = config.device
 
-    # model = BERTBaseUncased()
-    # # model.load_state_dict(torch.load(
-    # #     model_path, map_location=torch.device(device)))
-    # model.to(device)
+    model = BERTBaseUncased()
+    # model.load_state_dict(torch.load(
+    #     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)
 
     outputs =  classifier(sentence)