Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,6 @@ fine_tuned_model_name = "Vishwas1/bert-base-imdb" # Replace with your fine-tune
|
|
14 |
|
15 |
# Load models
|
16 |
original_tokenizer, original_model = load_model(original_model_name)
|
17 |
-
//fine_tuned_tokenizer, fine_tuned_model = load_model(fine_tuned_model_name)
|
18 |
fine_tuned_tokenizer = AutoTokenizer.from_pretrained("Vishwas1/bert-base-imdb")
|
19 |
fine_tuned_model = AutoModelForSequenceClassification.from_pretrained("Vishwas1/bert-base-imdb")
|
20 |
|
|
|
14 |
|
15 |
# Load models
|
16 |
original_tokenizer, original_model = load_model(original_model_name)
|
|
|
17 |
fine_tuned_tokenizer = AutoTokenizer.from_pretrained("Vishwas1/bert-base-imdb")
|
18 |
fine_tuned_model = AutoModelForSequenceClassification.from_pretrained("Vishwas1/bert-base-imdb")
|
19 |
|