Create app.py

app.py

@@ -0,0 +1,24 @@
+import os
+from fastapi import FastAPI
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
+
+app = FastAPI()
+
+MODEL_ID = "Adedayo2000/sentiment-model"
+
+hf_token = os.getenv("HF_TOKEN")
+
+print("Loading model...")
+tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, token=hf_token)
+model = AutoModelForSequenceClassification.from_pretrained(MODEL_ID, token=hf_token)
+print("Model loaded successfully!")
+
+@app.get("/")
+def read_root():
+    return {"message": "API is live!", "model": MODEL_ID}
+
+@app.post("/predict")
+def predict(text: str):
+    inputs = tokenizer(text, return_tensors="pt")
+    outputs = model(**inputs)
+    return {"logits": outputs.logits.tolist()}