Update app.py

app.py

@@ -8,6 +8,15 @@ app_description = """
 This app detects hate speech on Portuguese text using multiple models. You can either introduce your own sentences by filling in "Text" or click on one of the examples provided below.
 """
 
+app_examples = [
+    ["As pessoas tem que perceber que ser 'panasca' não é deixar de ser homem, é deixar de ser humano 😂😂"],
+    ["Isso pulhiticos merdosos, continuem a importar lixo, até Portugal deixar de ser Portugal."],
+    ["Vai pá puta que te pariu seu paneleiro do caralho, virgem ofendida"],
+    ["Vamo-nos unir para criar um mundo mais inclusivo e tolerante."],
+    ["Eu admiro muito a coragem e a determinação da minha colega de trabalho."],
+    ["O tempo está ensolarado hoje, perfeito para um passeio no parque."]
+]
+
 model_list = [
     "knowhate/HateBERTimbau",
     "knowhate/HateBERTimbau-youtube",
@@ -15,19 +24,20 @@ model_list = [
     "knowhate/HateBERTimbau-yt-tt",
 ]
 
+user_friendly_name = {
+    "knowhate/HateBERTimbau": "HateBERTimbau (Original)",
+    "knowhate/HateBERTimbau-youtube": "HateBERTimbau (YouTube)",
+    "knowhate/HateBERTimbau-twitter": "HateBERTimbau (Twitter)",
+    "knowhate/HateBERTimbau-yt-tt": "HateBERTimbau (YouTube + Twitter)",
+}
+
+reverse_user_friendly_name = { v:k for k,v in user_friendly_name.items() }
+user_friendly_name_list = list(user_friendly_name.values())
+
 #pipe = pipeline("text-classification", model="knowhate/HateBERTimbau")
 #demo = gr.Interface.from_pipeline(pipe)
 #demo.launch()
 
-app_examples = [
-    ["As pessoas tem que perceber que ser 'panasca' não é deixar de ser homem, é deixar de ser humano kkk"],
-    ["Isso pulhiticos merdosos, continuem a importar lixo, até Portugal deixar de ser Portugal."],
-    ["Vai pá puta que te pariu seu paneleiro do caralho, virgem ofendida"],
-    ["Vamo-nos unir para criar um mundo mais inclusivo e tolerante."],
-    ["Eu admiro muito a coragem e a determinação da minha colega de trabalho."],
-    ["O tempo está ensolarado hoje, perfeito para um passeio no parque."]
-]
-
 def predict(text, chosen_model):
 
     # Initialize the pipeline with the chosen model
@@ -39,7 +49,7 @@ def predict(text, chosen_model):
 
 inputs = [
     gr.Textbox(label="Text", value= app_examples[0][0]),
-    gr.Dropdown(label="Model", choices=model_list, value=model_list[2])
+    gr.Dropdown(label="Model", choices=user_friendly_name_list, value=user_friendly_name_list[2])
 ]
 
 outputs = [