sotosbarl commited on
Commit
46cd5e7
·
1 Parent(s): 0dc3afb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -8,12 +8,9 @@ model_name = "MoritzLaurer/mDeBERTa-v3-base-mnli-xnli"
8
  tokenizer = AutoTokenizer.from_pretrained(model_name)
9
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
10
 
11
- with open('articles_list.pkl', 'rb') as file:
12
- articles_list = pickle.load(file)
13
-
14
- label_names = []
15
- for i in articles_list[0:20]:
16
- label_names.append(i[0:15])
17
 
18
  def classify(text):
19
  input = tokenizer(text, truncation=True, return_tensors="pt")
 
8
  tokenizer = AutoTokenizer.from_pretrained(model_name)
9
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
10
 
11
+
12
+ label_names = ["γάμος", "αλλοδαπός", "φορολογία", "κληρονομικά", "στέγη", "οικογενειακό", "εμπορικό","κλοπή","απάτη"]
13
+
 
 
 
14
 
15
  def classify(text):
16
  input = tokenizer(text, truncation=True, return_tensors="pt")