Update app.py

app.py

@@ -7,6 +7,7 @@ from transformers import (
 )
 import torch
 import os
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
 hf_token = os.getenv("HF_TOKEN")
 
@@ -24,8 +25,19 @@ sentiment_model = AutoModelForSequenceClassification.from_pretrained(
     "EfektMotyla/absa-roberta", token=hf_token
 ).to(device)
 
-pl_to_en = pipeline("translation", model="Helsinki-NLP/opus-mt-pl-en", device=0 if device == "cuda" else -1)
-en_to_pl = pipeline("translation", model="gsarti/opus-mt-tc-en-pl", device=0 if device == "cuda" else -1)
+pl_to_en = pipeline(
+    "translation",
+    model="Helsinki-NLP/opus-mt-pl-en",
+    device=0 if device == "cuda" else -1,
+    token=hf_token
+)
+
+en_to_pl = pipeline(
+    "translation",
+    model="gsarti/opus-mt-tc-en-pl",
+    device=0 if device == "cuda" else -1,
+    token=hf_token
+)
 
 # === Dane wejściowe i wyjściowe ===
 class Comment(BaseModel):
@@ -84,6 +96,8 @@ def extract_aspects(text_en):
     return list(set([a.lower() for a in aspects]))
 
 # === Główna funkcja API ===
+app = FastAPI()
+
 @app.post("/analyze", response_model=AnalysisResult)
 def analyze_comment(comment: Comment):
     text_pl = comment.text