Updates

app.py

@@ -5,7 +5,7 @@ from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassifica
 models = ["bert-base-uncased", "roberta-base"]
 
 # Datasets included within the interface
-datasets = ["None",
+datasets = ["No Dataset Finetuning",
             "vedantgaur/GPTOutputs-MWP - AI Data Only",
             "vedantgaur/GPTOutputs-MWP - Human Data Only",
             "vedantgaur/GPTOutputs-MWP - Both AI and Human Data",
@@ -13,12 +13,12 @@ datasets = ["None",
 
 # Mapping of user-selected model and dataset to actual model name on Hugging Face
 model_mapping = {
-    ("bert-base-uncased", "None"): "bert-base-uncased",
+    ("bert-base-uncased", "No Dataset Finetuning"): "bert-base-uncased",
     ("bert-base-uncased", "vedantgaur/GPTOutputs-MWP - AI Data Only"): "SkwarczynskiP/bert-base-uncased-finetuned-vedantgaur-AI-generated",
     ("bert-base-uncased", "vedantgaur/GPTOutputs-MWP - Human Data Only"): "SkwarczynskiP/bert-base-uncased-finetuned-vedantgaur-human-generated",
     ("bert-base-uncased", "vedantgaur/GPTOutputs-MWP - Both AI and Human Data"): "SkwarczynskiP/bert-base-uncased-finetuned-vedantgaur-AI-and-human-generated",
     ("bert-base-uncased", "dmitva/human_ai_generated_text - Both AI and Human Data"): "SkwarczynskiP/bert-base-uncased-finetuned-dmitva-AI-and-human-generated",
-    ("roberta-base", "None"): "roberta-base",
+    ("roberta-base", "No Dataset Finetuning"): "roberta-base",
     ("roberta-base", "vedantgaur/GPTOutputs-MWP - AI Data Only"): "SkwarczynskiP/roberta-base-finetuned-vedantgaur-AI-generated",
     ("roberta-base", "vedantgaur/GPTOutputs-MWP - Human Data Only"): "SkwarczynskiP/roberta-base-finetuned-vedantgaur-human-generated",
     ("roberta-base", "vedantgaur/GPTOutputs-MWP - Both AI and Human Data"): "SkwarczynskiP/roberta-base-finetuned-vedantgaur-AI-and-human-generated",
@@ -47,10 +47,10 @@ def detect_ai_generated_text(model: str, dataset: str, text: str) -> str:
 
 # Examples included within the interface
 examples = [
-    ["AI Generated", ""],
-    ["Human Written", ""],
-    ["AI Generated", ""],
-    ["Human Written", ""]
+    ["AI Generated", "ex1"],
+    ["Human Written", "ex2"],
+    ["AI Generated", "ex3"],
+    ["Human Written", "ex4"]
 ]
 
 interface = gr.Interface(
@@ -62,7 +62,7 @@ interface = gr.Interface(
     ],
     outputs=gr.Textbox(label="Output"),
     examples=examples,
-    title="AI Generated Text Detection",
+    title="AI Generated Text Detection"
 )
 
 if __name__ == "__main__":