Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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-
|
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)
|