Update app.py

app.py

@@ -7,14 +7,19 @@ from transformers import (
 )
 import torch
 import os
+from pathlib import Path
+ROOT = Path(__file__).parent
+aspect_path    = ROOT / "models/bert-aspect-ner"
+sentiment_path = ROOT / "models/absa-roberta"
+
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-aspect_tokenizer = AutoTokenizer.from_pretrained("EfektMotyla/bert-aspect-ner")
-aspect_model = AutoModelForTokenClassification.from_pretrained("EfektMotyla/bert-aspect-ner").to(device)
 
-sentiment_tokenizer = AutoTokenizer.from_pretrained("EfektMotyla/absa-roberta")
-sentiment_model = AutoModelForSequenceClassification.from_pretrained("EfektMotyla/absa-roberta").to(device)
+aspect_tok  = AutoTokenizer.from_pretrained(aspect_path,  local_files_only=True)
+aspect_model= AutoModelForTokenClassification.from_pretrained(aspect_path, local_files_only=True).to(device)
 
+sent_tok  = AutoTokenizer.from_pretrained(sentiment_path,  local_files_only=True)
+sent_model= AutoModelForSequenceClassification.from_pretrained(sentiment_path, local_files_only=True).to(device)
 pl_to_en = pipeline(
     "translation",
     model="Helsinki-NLP/opus-mt-pl-en",