update

app.py

@@ -15,14 +15,12 @@ def classify(text):
         predicted_class = torch.argmax(logits, dim=1).item()
     return str(predicted_class)
 
-# Use Blocks
+# Create Gradio interface
 with gr.Blocks() as demo:
     txt_in = gr.Textbox(label="Enter text")
     txt_out = gr.Textbox(label="Prediction")
     btn = gr.Button("Classify")
     btn.click(classify, txt_in, txt_out)
 
-    # ✅ Enable a predictable API endpoint
-    demo.load(fn=classify, inputs=txt_in, outputs=txt_out, api_name="predict")
-
-demo.launch()
+# Launch with API enabled
+demo.launch()