Update app.py

app.py

@@ -1,18 +1,20 @@
+from fastapi import FastAPI
 from transformers import pipeline
 import gradio as gr
 
-# Load model with CPU (required for free Spaces)
-model = pipeline("text-classification", 
-                model="win2win/3-epochs-classifier-ver2",
-                device="cpu")
+app = FastAPI()
+model = pipeline("text-classification", model="win2win/3-epochs-classifier-ver2")
 
+# Custom endpoint
+@app.post("/predict")
+async def api_predict(text: str):
+    return model(text)
+
+# Gradio interface
 def predict(text):
     return model(text)
 
-# Create Gradio interface
-gr.Interface(
-    fn=predict,
-    inputs="text",
-    outputs="json",
-    title="Text Classifier"
-).launch()
+io = gr.Interface(fn=predict, inputs="text", outputs="json")
+
+# Mount both endpoints
+app = gr.mount_gradio_app(app, io, path="/")