Update app.py

app.py

@@ -1,17 +1,24 @@
-import gradio as gr
 import os
+from fastapi import FastAPI, Request
+from fastapi.responses import JSONResponse
 from transformers import pipeline
 
+app = FastAPI()
+
 token = os.environ.get("HF_TOKEN")
 
 english_model = pipeline("text-classification", model="mrgmd01/Finetuned_siebert-sentiment-roberta-large-english", token=token)
 urdu_model = pipeline("text-classification", model="mrgmd01/SA_Model_bert-base-multilingual-uncased", token=token)
 
-def predict(text, language):
-    if language == "urdu" or language == "roman_urdu":
+@app.post("/predict")
+async def predict(request: Request):
+    body = await request.json()
+    text = body.get("text", "")
+    language = body.get("language", "english")
+    
+    if language in ["urdu", "roman_urdu"]:
         result = urdu_model(text)[0]
     else:
         result = english_model(text)[0]
-    return {"label": result["label"], "score": result["score"]}
-
-gr.Interface(fn=predict, inputs=["text", "text"], outputs="json", api_name="predict").launch()
+    
+    return JSONResponse({"label": result["label"], "score": result["score"]})