Update app.py

app.py

@@ -67,16 +67,22 @@ model = AutoModelForSequenceClassification.from_pretrained(model_name).to(device
 tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
 
 # sentiment model - add no_cache=True and force_download=True
+# Model initialization
+model_name = "SamanthaStorm/tether-multilabel-v4"
+model = AutoModelForSequenceClassification.from_pretrained(model_name).to(device)
+tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
+
+# sentiment model
 sentiment_model = AutoModelForSequenceClassification.from_pretrained(
     "SamanthaStorm/tether-sentiment", 
-    no_cache=True,
-    force_download=True
+    force_download=True,
+    local_files_only=False
 ).to(device)
 sentiment_tokenizer = AutoTokenizer.from_pretrained(
     "SamanthaStorm/tether-sentiment", 
     use_fast=False,
-    no_cache=True,
-    force_download=True
+    force_download=True,
+    local_files_only=False
 )
 
 # After loading the sentiment model
@@ -85,6 +91,7 @@ logger.debug(f"Model name: {sentiment_model.config.name_or_path}")
 logger.debug(f"Last modified: {sentiment_model.config._name_or_path}")
 
 
+
 emotion_pipeline = hf_pipeline(
     "text-classification",
     model="j-hartmann/emotion-english-distilroberta-base",