Update app.py

app.py

@@ -47,7 +47,7 @@ exampleDatasets = ["No Dataset Finetuning",
 examples = [[random.choice(exampleModels), random.choice(exampleDatasets), random.choice(exampleText)] for example in exampleText]
 
 
-def detect_ai_generated_text(model: str, dataset: str, text: str) -> dict:
+def detect_ai_generated_text(model: str, dataset: str, text: str) -> list:
     # Get the fine-tuned model using mapping
     finetuned_model = model_mapping.get((model, dataset))
 
@@ -69,7 +69,7 @@ def detect_ai_generated_text(model: str, dataset: str, text: str) -> dict:
     ai_bar = f'<div style="background-color: red; width: {ai_score * 100}%; height: 20px;"></div>'
     human_bar = f'<div style="background-color: blue; width: {human_score * 100}%; height: 20px;"></div>'
 
-    return {"label": f"{label} with confidence {score * 100:.2f}%", "ai_bar": ai_bar, "human_bar": human_bar}
+    return [f"{label} with confidence {score * 100:.2f}%", ai_bar, human_bar]
 
 
 interface = gr.Interface(