Update app.py

app.py

@@ -1,37 +1,23 @@
-from fastapi import FastAPI
-from pydantic import BaseModel
-from src.models.inference import SentimentModel
-from fastapi.middleware.cors import CORSMiddleware
 import nltk
 nltk.download('stopwords')
 
-app = FastAPI()
+import gradio as gr
+from src.models.inference import SentimentModel
 
 model = SentimentModel()
 
+def predict(text):
+    result = model.predict(text)
+    return result
 
-class TextInput(BaseModel):
-    text: str
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],  # later restrict this
-    allow_methods=["*"],
-    allow_headers=["*"],
+demo = gr.Interface(
+    fn=predict,
+    inputs=gr.Textbox(label="Enter Text"),
+    outputs=gr.Label(label="Sentiment"),
+    title="Sentiment Analysis",
 )
 
-@app.get("/")
-def home():
-    return {"message": "Sentiment API running"}
-
-
-@app.post("/predict")
-def predict(input: TextInput):
-    result = model.predict(input.text)
-
-    return {
-        "text": input.text,
-        "sentiment": result
-    }
+demo.launch()