Vishwas1 commited on
Commit
a7bf09d
·
verified ·
1 Parent(s): 04919db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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, original_model = load_model(original_model_name)
 
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