Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -36,8 +36,8 @@ def load_model():
|
|
36 |
|
37 |
# Load T5 for Paraphrasing
|
38 |
|
39 |
-
|
40 |
-
t5_model = export_and_get_onnx_model('valurank/t5-paraphraser')
|
41 |
t5_tokenizer = T5Tokenizer.from_pretrained('t5-base')
|
42 |
t5_model = t5_model.to(device)
|
43 |
return kw_extractor, t5_model, t5_tokenizer
|
|
|
36 |
|
37 |
# Load T5 for Paraphrasing
|
38 |
|
39 |
+
t5_model = T5ForConditionalGeneration.from_pretrained('valurank/t5-paraphraser', use_auth_token=True)
|
40 |
+
#t5_model = export_and_get_onnx_model('valurank/t5-paraphraser')
|
41 |
t5_tokenizer = T5Tokenizer.from_pretrained('t5-base')
|
42 |
t5_model = t5_model.to(device)
|
43 |
return kw_extractor, t5_model, t5_tokenizer
|