app

app.py

@@ -11,7 +11,7 @@ def classify_text(input_text,id2label):
     inputs = tokenizer(input_text, return_tensors="pt")
     outputs = model(**inputs)
     pred = np.argmax(outputs)
-    return(f"Predicted class: {id2label[pred]}")
+    return(pred)
 
 demo = gr.Interface(classify_text, "text", "text")
 demo.launch()