Vishwas1 commited on
Commit
37412ce
·
verified ·
1 Parent(s): 9890682

Update app.py

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