Update app.py

app.py

@@ -15,6 +15,6 @@ def classify_text(text):
     predicted_class = logits.argmax().item()  # Get the predicted class
     return f"Predicted class: {predicted_class}"
 
-# Gradio Interface
-demo = gr.Interface(fn=classify_text, inputs="text", outputs="text", layout="vertical")
+# Gradio Interface without forced layout
+demo = gr.Interface(fn=classify_text, inputs="text", outputs="text")
 demo.launch()