Update app.py

app.py

@@ -12,9 +12,9 @@ from data_predict import convert_to_stsb_features,convert_to_features
 from huggingface_hub import hf_hub_download,snapshot_download
 
 
-features_dict = {}
-extra_feature_dict = {}
-sentinews_location = ""
+# features_dict = {}
+# extra_feature_dict = {}
+# sentinews_location = ""
 
 # df_document_croatian_test = pd.read_csv(sentinews_location+"textlabel.tsv", sep="\t")
 # df_document_croatian_test = df_document_croatian_test[["content"]]
@@ -43,7 +43,8 @@ def predict():
     }
     
     features_dict = convert_to_features(dataset_dict, convert_func_dict)
-
+    
+    return features_dict
 
 
 #model_link = snapshot_download(repo_id="FFZG-cleopatra/Croatian-News-Classifier")
@@ -54,6 +55,7 @@ multitask_model.load_state_dict(torch.load(model_link, map_location=torch.device
 # device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # multitask_model.to(device)
 predictions = []
+features_dict = predict()
 for _, batch in enumerate(features_dict["document"]['test']):
     for key, value in batch.items():
         batch[key] = batch[key].to(device)