Update app.py

app.py

@@ -16,10 +16,13 @@ name = "karalif/myTestModel"
 model = AutoModelForSequenceClassification.from_pretrained(name)
 tokenizer = AutoTokenizer.from_pretrained(name, normalization=True)
 
-text = "hvað er maðurinn eiginlega að pæla ég fatta ekki??????????"
-tokens = tokenizer.tokenize(text)
-decoded_text = tokenizer.convert_tokens_to_string(tokens)
 bench = Benchmark(model, tokenizer)
+text = "hvað er maðurinn eiginlega að pæla ég fatta ekki??????????"
+
+explanations_formality = bench.explain(text, target=0)
+explanations_sentiment = bench.explain(text, target=1)
+explanations_politeness = bench.explain(text, target=2)
+explanations_toxicity = bench.explain(text, target=3)
 
 def get_prediction(text):
     encoding = tokenizer(text, return_tensors="pt", padding="max_length", truncation=True, max_length=200)