Update app.py

app.py

@@ -1,12 +1,10 @@
 import gradio as gr
-import torch
 from transformers import pipeline
 
 # Load the model
 model_name = "knowledgator/comprehend_it-base"
 classifier = pipeline("zero-shot-classification", model=model_name, device="cpu")
 
-
 # Function to classify feedback
 def classify_feedback(feedback_text):
     # Classify feedback using the loaded model
@@ -20,7 +18,7 @@ def classify_feedback(feedback_text):
     return {top_labels[i]: scores[i] for i in range(len(top_labels))}
 
 # Create Gradio interface
-feedback_textbox = gr.inputs.Textbox(lines=5, label="Enter your feedback:")
+feedback_textbox = gr.inputs.Text(label="Enter your feedback:")
 feedback_output = gr.outputs.Textbox(label="Top 2 Labels with Scores:")
 
 gr.Interface(