Update app.py

app.py

@@ -7,24 +7,17 @@ def install(package):
 install("transformers")
 install("torch")
 
-from transformers import pipeline
 import gradio as gr
 
-model = pipeline("text-classification", model="win2win/3-epochs-classifier-ver2")
-
-def predict(text):
-    return model(text)
-
-gr.Interface(fn=predict, inputs="text", outputs="json").launch()
-
-import gradio as gr
+from fastapi import FastAPI
 from transformers import pipeline
 
+app = FastAPI()
 
-
+# Load your Hugging Face model
 model = pipeline("text-classification", model="win2win/3-epochs-classifier-ver2")
 
-def predict(text):
-    return model(text)
-
-gr.Interface(fn=predict, inputs="text", outputs="json").launch()
+# Define a POST endpoint at `/predict`
+@app.post("/predict")
+async def predict(text: str):
+    return model(text)