Update app.py

app.py

@@ -6,12 +6,6 @@ import os
 
 # Load the original pre-trained model
 def load_model(model_name):
-    hf_token = os.getenv('HUGGINGFACE_TOKEN')
-    if not hf_token:
-        return "Error: Hugging Face token not found. Please set it as a secret."
-    
-    # Login to Hugging Face Hub
-    login(hf_token)
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForSequenceClassification.from_pretrained(model_name)
     return tokenizer, model
@@ -21,6 +15,12 @@ original_model_name = "Vishwas1/hummingbird-base-marathi"  # Replace with your o
 fine_tuned_model_name = "Vishwas1/hummingbird-finetuned-marathi"  # Replace with your fine-tuned model's repo ID
 
 # Load models
+hf_token = os.getenv('HUGGINGFACE_TOKEN')
+    if not hf_token:
+        return "Error: Hugging Face token not found. Please set it as a secret."
+    
+# Login to Hugging Face Hub
+login(hf_token)
 original_tokenizer = AutoTokenizer.from_pretrained(original_model_name)
 original_model = AutoModelForSequenceClassification.from_pretrained(original_model_name)
 fine_tuned_tokenizer = AutoTokenizer.from_pretrained(fine_tuned_model_name)