Vishwas1 commited on
Commit
b1a3c74
·
verified ·
1 Parent(s): 6134b21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ def load_model(model_name):
10
 
11
  # Models to compare
12
  original_model_name = "bert-base-uncased" # Replace with your original model
13
- fine_tuned_model_name = "Vishwas1/bert-base-imdb2" # Replace with your fine-tuned model's repo ID
14
 
15
  # Load models
16
  original_tokenizer, original_model = load_model(original_model_name)
 
10
 
11
  # Models to compare
12
  original_model_name = "bert-base-uncased" # Replace with your original model
13
+ fine_tuned_model_name = "Vishwas1/bert-base-imdb3" # Replace with your fine-tuned model's repo ID
14
 
15
  # Load models
16
  original_tokenizer, original_model = load_model(original_model_name)