Update README.md

README.md

@@ -1,28 +1,23 @@
 ---
-language: "multilingual"
+language: "it"
 tags:
 - bert
 - sarcasm-detection
 - text-classification
 widget:
-- text: "CIA Realizes It's Been Using Black Highlighters All These Years."
+- text: "Auto, stop a diesel e benzina dal 2035. Ecco cosa cambia per i consumatori"
 ---
 
-# Multilingual Sarcasm Detector
+# Italian Sarcasm Detector
 
-Multilingual Sarcasm Detector is a text classification model built to detect sarcasm from news article titles. It is fine-tuned on bert-multilingual uncased and the training data consists of ready-made datasets available on Kaggle as well scraped data from multiple newspapers in English, Dutch and Italian.
+Italian Sarcasm Detector is a text classification model built to detect sarcasm from news article titles. It is fine-tuned on dbmdz/bert-base-italian-uncased and the training data consists of scraped data from Italian non-sarcastic newspaper (Il Giornale) and sarcastic newspaper (Lercio).
 
 ## Metrics:
 
 
 ## Training Data
 
-Datasets:
-- English language data: [Kaggle: News Headlines Dataset For Sarcasm Detection]([https://www.kaggle.com/datasets/rmisra/news-headlines-dataset-for-sarcasm-detection]).
-- Dutch non-sarcastic data: [Kaggle: Dutch News Articles]([https://www.kaggle.com/datasets/maxscheijen/dutch-news-articles])
-
 Scraped data:
-- Dutch sarcastic news from [De Speld]([https://speld.nl])
 - Italian non-sarcastic news from [Il Giornale]([https://www.ilgiornale.it])
 - Italian sarcastic news from [Lercio]([https://www.lercio.it])
 
@@ -41,12 +36,12 @@ import string
 def preprocess_data(text: str) -> str:
    return text.lower().translate(str.maketrans("", "", string.punctuation)).strip()
 
-MODEL_PATH = "helinivan/multilingual-sarcasm-detector"
+MODEL_PATH = "helinivan/italian-sarcasm-detector"
 
 tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
 model = AutoModelForSequenceClassification.from_pretrained(MODEL_PATH)
 
-text = "CIA Realizes It's Been Using Black Highlighters All These Years."
+text = "Auto, stop a diesel e benzina dal 2035. Ecco cosa cambia per i consumatori"
 tokenized_text = tokenizer([preprocess_data(text)], padding=True, truncation=True, max_length=512, return_tensors="pt")
 output = model(**tokenized_text)
 probs = output.logits.softmax(dim=-1).tolist()[0]
@@ -66,4 +61,5 @@ Output:
 | Model-Name | F1 | Precision | Recall | Accuracy 
 | ------------- |:-------------| -----| -----| ----| 
 | helinivan/english-sarcasm-detector | 94.48 | 94.46 | 94.51 | 94.48
+| helinivan/italian-sarcasm-detector | 92.99 | 92.77 | 93.24 | 93.42
 | helinivan/multilingual-sarcasm-detector | 90.91 | 91.51 | 90.44 | 91.55