Update main.py

main.py

@@ -1,11 +1,16 @@
+import os
 from fastapi import FastAPI
 from pydantic import BaseModel
 from transformers import pipeline
 
+# Set custom cache directory to avoid permission issues
+os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
+
 app = FastAPI()
 
-# Load sentiment analysis model
-sentiment_pipeline = pipeline("sentiment-analysis")
+# Explicitly specify a model (avoid default selection)
+MODEL_NAME = "distilbert-base-uncased-finetuned-sst-2-english"
+sentiment_pipeline = pipeline("sentiment-analysis", model=MODEL_NAME)
 
 class SentimentRequest(BaseModel):
     text: str