From 54654ced401b3cc347a90cc8cb358436c1edacf6 Mon Sep 17 00:00:00 2001
From: Rob Caamano <65639885+RobCaamano@users.noreply.github.com>
Date: Sun, 23 Apr 2023 18:37:43 -0400
Subject: [PATCH 1/2] Update README.md

---
 README.md | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/README.md b/README.md
index f493961..4175414 100644
--- a/README.md
+++ b/README.md
@@ -10,3 +10,5 @@ pinned: false
 ---
 
 # Finetuning_Language_Models-Toxic_Tweets
+
+URL: https://huggingface.co/spaces/RobCaamano/Finetuning_Language_Models-Toxic_Tweets

From 55a1776f23eca7d217a26347f6ba3086fdcd65e5 Mon Sep 17 00:00:00 2001
From: Rob Caamano <65639885+RobCaamano@users.noreply.github.com>
Date: Mon, 24 Apr 2023 16:11:02 -0400
Subject: [PATCH 2/2] App - Implemented RObert Model

---
 app.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/app.py b/app.py
index 4286913..e86be00 100644
--- a/app.py
+++ b/app.py
@@ -14,6 +14,7 @@
 model_options = {
     "DistilBERT Base Uncased (SST-2)": "distilbert-base-uncased-finetuned-sst-2-english",
     "Fine-tuned Toxicity Model": "RobCaamano/toxicity_distilbert",
+    "Fine-tuned Toxicity Model (RObert)": "RobCaamano/toxicity_RObert",  # Added new model option
 }
 selected_model = st.selectbox("Select Model", options=list(model_options.keys()))
 
@@ -22,7 +23,7 @@
 tokenizer = AutoTokenizer.from_pretrained(mod_name)
 model = AutoModelForSequenceClassification.from_pretrained(mod_name)
 
-if selected_model == "Fine-tuned Toxicity Model":
+if selected_model in ["Fine-tuned Toxicity Model", "Fine-tuned Toxicity Model (RObert)"]:
     toxicity_classes = ["toxic", "severe_toxic", "obscene", "threat", "insult", "identity_hate"]
     model.config.id2label = {i: toxicity_classes[i] for i in range(model.config.num_labels)}
 
@@ -38,7 +39,7 @@ def get_toxicity_class(prediction):
     
     tweet_portion = text[:50] + "..." if len(text) > 50 else text
 
-    if selected_model == "Fine-tuned Toxicity Model":
+    if selected_model in ["Fine-tuned Toxicity Model", "Fine-tuned Toxicity Model (RObert)"]:
         column_name = "Toxicity Class"
     else:
         column_name = "Prediction"