updated app.py

app.py

@@ -32,7 +32,7 @@ def moderate_image(img):
 
 def classify_toxic(text):
     batch = tokenizer.encode(text, return_tensors="pt")
-    output = model(batch).cpu()
+    output = model(batch)
     output = output.tolist()
     max_idx = max(output)
     return "Toxic" if output.index(max_idx) == 1 else "Neutral"
@@ -151,7 +151,7 @@ with gr.Blocks(
                 with gr.Row():
                     with gr.Column(scale=3):
                         with gr.Group(elem_classes="apple-card"):
-                            txt_in = gr.Textbox(lines=6, label="Enter Text")
+                            txt_in = gr.Textbox(lines=4, label="Enter Text")
                             classify_txt_btn = gr.Button("Analyze")
                             text_clear_btn = gr.ClearButton(components=txt_in)