Added in more finetuned models

app.py

@@ -1,16 +1,25 @@
 import gradio as gr
 from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
 
-
-models = ["bert-base-uncased", "model2"]
-datasets = ["vedantgaur/GPTOutputs-MWP", "dataset2"]
+models = ["bert-base-uncased", "roberta-base"]
+datasets = ["None", 
+            "vedantgaur/GPTOutputs-MWP - AI Data Only",
+            "vedantgaur/GPTOutputs-MWP - Human Data Only",
+            "vedantgaur/GPTOutputs-MWP - Both AI and Human Data",
+            "dmitva/human_ai_generated_text - Both AI and Human Data"]
 
 # Mapping of user-selected model and dataset to actual model name on Hugging Face
 model_mapping = {
-    ("bert-base-uncased", "vedantgaur/GPTOutputs-MWP"): "SkwarczynskiP/bert-base-uncased-finetuned-vedantgaur",
-    ("bert-base-uncased", "dataset2"): "finetuned_model1_on_dataset2",
-    ("model2", "vedantgaur/GPTOutputs-MWP"): "finetuned_model2_on_dataset1",
-    ("model2", "dataset2"): "finetuned_model2_on_dataset2",
+    ("bert-base-uncased", "None"): "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"): "SkwarczynskiP/bert-base-uncased-finetuned-dmitva-AI-and-human-generated",
+    ("roberta-base", "None"): "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",
+    ("roberta-base", "dmitva/human_ai_generated_text"): "SkwarczynskiP/roberta-base-finetuned-dmitva-AI-and-human-generated"
 }
 
 def detect_ai_generated_text(model: str, dataset: str, text: str) -> str: