update app.py 2

app.py

@@ -1,9 +1,4 @@
-import gradio as gr
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
-import torch
-
-# Load model
-model_id = "Rerandaka/Cild_safety_bigbird"
+d = "Rerandaka/Cild_safety_bigbird"
 tokenizer = AutoTokenizer.from_pretrained(model_id, use_fast=False)
 model = AutoModelForSequenceClassification.from_pretrained(model_id)
 
@@ -15,12 +10,13 @@ def classify(text):
         predicted_class = torch.argmax(logits, dim=1).item()
     return str(predicted_class)
 
-# 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)
+# ✅ Create API-ready interface
+demo = gr.Interface(
+    fn=classify,
+    inputs=gr.Textbox(label="Enter text"),
+    outputs=gr.Textbox(label="Prediction")
+)
 
-# Launch with API enabled
-demo.launch()
+# ✅ Enable queue and API support
+demo.queue()              # <-- Enables /queue/join
+demo.launch(show_api=True)  # <-- Enables gradio_client + /predict