Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,8 @@ original_model_name = "Vishwas1/hummingbird-base-marathi" # Replace with your o
|
|
21 |
fine_tuned_model_name = "Vishwas1/hummingbird-finetuned-marathi" # Replace with your fine-tuned model's repo ID
|
22 |
|
23 |
# Load models
|
24 |
-
original_tokenizer
|
|
|
25 |
fine_tuned_tokenizer = AutoTokenizer.from_pretrained(fine_tuned_model_name)
|
26 |
fine_tuned_model = AutoModelForSequenceClassification.from_pretrained(fine_tuned_model_name)
|
27 |
|
|
|
21 |
fine_tuned_model_name = "Vishwas1/hummingbird-finetuned-marathi" # Replace with your fine-tuned model's repo ID
|
22 |
|
23 |
# Load models
|
24 |
+
original_tokenizer = AutoTokenizer.from_pretrained(original_model_name)
|
25 |
+
original_model = AutoModelForSequenceClassification.from_pretrained(original_model_name)
|
26 |
fine_tuned_tokenizer = AutoTokenizer.from_pretrained(fine_tuned_model_name)
|
27 |
fine_tuned_model = AutoModelForSequenceClassification.from_pretrained(fine_tuned_model_name)
|
28 |
|