arithescientist commited on
Commit
176379b
·
1 Parent(s): db36091

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -54,7 +54,7 @@ custom_config.output_hidden_states=True
54
  print("lets go")
55
  custom_tokenizer = AutoTokenizer.from_pretrained(model_name)
56
  print("lets go")
57
- #custom_model = AutoModel.from_pretrained(model_name, config=custom_config)
58
  print("lets go")
59
  bert_legal_model = Summarizer(custom_model=custom_model, custom_tokenizer=custom_tokenizer)
60
  print('Using model {}\n'.format(model_name))
 
54
  print("lets go")
55
  custom_tokenizer = AutoTokenizer.from_pretrained(model_name)
56
  print("lets go")
57
+ custom_model = AutoModel.from_pretrained(model_name, config=custom_config)
58
  print("lets go")
59
  bert_legal_model = Summarizer(custom_model=custom_model, custom_tokenizer=custom_tokenizer)
60
  print('Using model {}\n'.format(model_name))