AnushkaTonk commited on
Commit
d43bb10
·
1 Parent(s): ec31de8

updated load_model_and_tokenizer() in app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -43,8 +43,8 @@ models_info = {
43
  # loading models and tokenizers:
44
  def load_model_and_tokenizer(model_choice):
45
  config = models_info[model_choice]
46
- tokenizer = config["tokenizer"].from_pretrained(config["path"], local_files_only=True)
47
- model = config["model_class"].from_pretrained(config["path"], local_files_only=True)
48
  try:
49
  id2label = model.config.id2label
50
  if "LABEL_0" in id2label.values():
 
43
  # loading models and tokenizers:
44
  def load_model_and_tokenizer(model_choice):
45
  config = models_info[model_choice]
46
+ tokenizer = config["tokenizer"].from_pretrained(config["path"])
47
+ model = config["model_class"].from_pretrained(config["path"])
48
  try:
49
  id2label = model.config.id2label
50
  if "LABEL_0" in id2label.values():