Update app.py

app.py

@@ -129,29 +129,18 @@ def predict(image: Image.Image, text: str) -> str:
         )
         predicted_class = torch.sigmoid(classification_output).round().item()
     
-    return "Fake News" if predicted_class == 1 else "Real News"
-
-# Custom Theme
-theme = gr.themes.Base(
-    primary_hue="slate",
-    secondary_hue="blue",
-    neutral_hue="gray",
-    text_size="lg",
-    radius_size="md",
-    font=["Roboto", "sans-serif"]
-)
+    return "Biased" if predicted_class == 1 else "Unbiased"
 
 # Gradio Interface
 interface = gr.Interface(
     fn=predict,
     inputs=[
-        gr.Image(type="pil", label="Upload Related Image"),
-        gr.Textbox(lines=2, placeholder="Enter news text for classification...", label="Input Text")
+        gr.Image(type="pil", label="Upload Image"),
+        gr.Textbox(lines=2, placeholder="Enter text for classification...", label="Input Text")
     ],
     outputs=gr.Label(label="Prediction"),
-    title="Fake News Detector",
-    description="Upload an image and provide text to classify the news as 'Fake' or 'Real'.",
-    theme=theme
+    title="Multimodal Bias Classifier",
+    description="Upload an image and provide a text to classify it as 'Biased' or 'Unbiased'."
 )
 
 interface.launch()