Update app.py

app.py

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