Update app.py

app.py

@@ -14,7 +14,7 @@ fine_tuned_model_name = "Vishwas1/bert-base-imdb"  # Replace with your fine-tune
 
 # Load models
 original_tokenizer, original_model = load_model(original_model_name)
-fine_tuned_tokenizer = AutoTokenizer.from_pretrained("Vishwas1/bert-base-imdb")
+fine_tuned_tokenizer = AutoTokenizer.from_pretrained(original_model_name)
 fine_tuned_model = AutoModelForSequenceClassification.from_pretrained("Vishwas1/bert-base-imdb")
 
 # Ensure models are in evaluation mode