UI - theme

app.py

@@ -25,31 +25,31 @@ label_mapping = {
 }
 
 def classify_text(text):
-    inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
+    inputs = tokenizer(text, return_tensors="pt", truncation=True)
     inputs = {key: value.to(device) for key, value in inputs.items()}
     
     with torch.no_grad():
         outputs = model(**inputs)
         probabilities = torch.softmax(outputs.logits, dim=1)[0]
         predicted_class = torch.argmax(probabilities).item()
-        confidence = probabilities[predicted_class].item()
-        
+        confidence = probabilities[predicted_class].item() * 100
+
         if predicted_class == 24:
-            prediction_label = "✅ **Human Written**"
-            confidence_message = f"🔒 **Confidence:** {confidence:.2f}"
-            if confidence > 0.8:
-                confidence_message += " (Highly Likely Human)"
+            prediction_label = f"✅ The text is **{confidence:.2f}%** likely **Human written**."
+            model_info = ""
         else:
-            prediction_label = f"🤖 **AI Generated by {label_mapping[predicted_class]}**"
-            confidence_message = f"🔒 **Confidence:** {confidence:.2f}"
-            if confidence > 0.8:
-                confidence_message += " (Highly Likely AI)"
-    
-    return f"**Result:**\n\n{prediction_label}\n\n{confidence_message}"
+            prediction_label = f"🤖 The text is **{confidence:.2f}%** likely **AI generated**."
+            model_info = f"**Identified AI Model:** {label_mapping[predicted_class]}"
 
-title = "🧠 SzegedAI ModernBERT Text Detector"
+    result_message = f"**Result:**\n\n{prediction_label}"
+    if model_info:
+        result_message += f"\n\n{model_info}"
+        
+    return result_message
+
+title = "AI Generated Text Detector"
 description = """
-**AI Detection Tool by SzegedAI**
+**AI detection tool by SzegedAI**
 
 Detect AI-generated texts with precision using the new **ModernBERT** model, fine-tuned for machine-generated text detection, and capable of identifying 40 different models.
 
@@ -62,22 +62,18 @@ Detect AI-generated texts with precision using the new **ModernBERT** model, fin
 iface = gr.Interface(
     fn=classify_text,
     inputs=gr.Textbox(
-        label="✏️ Enter Text for Analysis", 
+        label="Enter Text for Analysis", 
         placeholder="Type or paste your content here...",
         lines=5,
         elem_id="text_input_box"
     ),
-    outputs=gr.Textbox(
-        label="Detection Results", 
-        lines=4,
-        elem_id="result_output_box"
-    ),
+    outputs=gr.Markdown(),
     title=title,
     description=description,
     allow_flagging="never", 
     live=False,  
     css="""
-    #text_input_box, #result_output_box {
+    #text_input_box {
         border-radius: 10px;
         border: 2px solid #4CAF50;
         font-size: 18px;