Update app.py

app.py

@@ -8,6 +8,7 @@ model = AutoModelForSequenceClassification.from_pretrained("willco-afk/my-model-
 
 # Function to classify input text
 def classify_text(text):
+    print("Classifying:", text)  # Check if this gets printed
     inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
     with torch.no_grad():
         logits = model(**inputs).logits
@@ -15,5 +16,5 @@ def classify_text(text):
     return f"Predicted class: {predicted_class}"
 
 # Gradio Interface
-demo = gr.Interface(fn=classify_text, inputs="text", outputs="text")
+demo = gr.Interface(fn=classify_text, inputs="text", outputs="text", layout="vertical")
 demo.launch()