Update app.py

app.py

@@ -13,7 +13,6 @@ else:
     device = torch.device("cpu")
 
 dataset_path = './codice_civile_ITA_LIBRI_2_withArtRef_v2.csv'
-input_model_path = './MODELLO_LOCALE_LIBRI_2_v5_2_subset60UniRRemphT4'
 
 
 def load_CC_from_CSV(path):
@@ -34,8 +33,8 @@ def load_CC_from_CSV(path):
 
 article_id, id_article, article_text, NUM_ART = load_CC_from_CSV(dataset_path)
 
-model = BertForSequenceClassification.from_pretrained(input_model_path)
-tokenizer = BertTokenizerFast.from_pretrained(input_model_path)
+model = BertForSequenceClassification.from_pretrained("AndreaSimeri/LamBERTa_v5")
+tokenizer = BertTokenizerFast.from_pretrained("AndreaSimeri/LamBERTa_v5")
 
 def LamBERTa_v5_placeholder(query):
     n = 345