Add application file

app.py

@@ -1,8 +1,12 @@
+import os
 from fastapi import FastAPI, Query
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from scipy.special import softmax
 from typing import Dict
 
+# Set TRANSFORMERS_CACHE to a writable directory
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/.cache"
+
 app = FastAPI()
 
 # Load model and tokenizer
@@ -34,7 +38,3 @@ def analyze_sentiment(tweet: str) -> Dict[str, float]:
 async def analyze_sentiment_endpoint(tweet: str = Query(..., description="The tweet to analyze")):
     sentiment_scores = analyze_sentiment(tweet)
     return {"sentiment": sentiment_scores}
-
-# To run the application:
-# 1. Save this code as `main.py`.
-# 2. Run the command `uvicorn main:app --reload`.

requirements.txt

@@ -3,4 +3,8 @@ uvicorn[standard]
 transformers[standard]
 torch[standard]
 torchvision[standard]
-torchaudio[standard]
+torchaudio[standard]
+fastapi
+uvicorn
+transformers
+scipy