feat: changing model again roBERTa -> longformer

app.py

@@ -7,10 +7,10 @@ from typing import List
 # Initialize FastAPI
 app = FastAPI()
 
-# Load RoBERTa sentiment analysis model
+# Load Longformer sentiment analysis model
 sentiment_pipeline = pipeline(
     "sentiment-analysis", 
-    model="cardiffnlp/twitter-roberta-base-sentiment"
+    model="allenai/longformer-base-4096"
 )
 
 # Request models
@@ -28,7 +28,7 @@ class SentimentResponse(BaseModel):
     sentiment_label: str
     confidence: float
 
-# Mapping RoBERTa labels to a floating-point scale
+# Mapping Longformer labels to a floating-point scale
 LABEL_MAP = {
     "LABEL_0": -1.0,  # Negative
     "LABEL_1": 0.0,   # Neutral
@@ -43,7 +43,7 @@ def analyze_sentiment(request: SentimentRequest):
         label = result["label"]
         score = result["score"]
         
-        # Convert RoBERTa labels to floating-point scores
+        # Convert Longformer labels to floating-point scores
         sentiment_score = LABEL_MAP[label]
         confidence = round(score, 3)
         
@@ -65,7 +65,7 @@ def analyze_sentiment_batch(request: BatchSentimentRequest):
             label = result["label"]
             score = result["score"]
             
-            # Convert RoBERTa labels to floating-point scores
+            # Convert Longformer labels to floating-point scores
             sentiment_score = LABEL_MAP[label]
             confidence = round(score, 3)
             
@@ -82,4 +82,4 @@ def analyze_sentiment_batch(request: BatchSentimentRequest):
 # Root endpoint
 @app.get("/")
 def root():
-    return {"message": "RoBERTa Sentiment Analysis API is running!"}
+    return {"message": "Longformer Sentiment Analysis API is running!"}