Spaces:
Running
Running
arithescientist
commited on
Commit
·
176379b
1
Parent(s):
db36091
Update app.py
Browse files
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 |
-
|
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))
|