update app.py 3

app.py

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