Update app.py

app.py

@@ -10,12 +10,12 @@ def load_model(model_name):
 
 # Models to compare
 original_model_name = "bert-base-uncased"  # Replace with your original model
-fine_tuned_model_name = "Vishwas1/bert-base-imdb"  # Replace with your fine-tuned model's repo ID
+fine_tuned_model_name = "Vishwas1/bert-base-imdb2"  # Replace with your fine-tuned model's repo ID
 
 # Load models
 original_tokenizer, original_model = load_model(original_model_name)
-fine_tuned_tokenizer = AutoTokenizer.from_pretrained(original_model_name)
-fine_tuned_model = AutoModelForSequenceClassification.from_pretrained("Vishwas1/bert-base-imdb")
+fine_tuned_tokenizer = AutoTokenizer.from_pretrained(fine_tuned_model_name)
+fine_tuned_model = AutoModelForSequenceClassification.from_pretrained(fine_tuned_model_name)
 
 # Ensure models are in evaluation mode
 original_model.eval()