Spaces:

jaimin
/

FastT5

Runtime error

jaimin commited on Oct 5, 2022

Commit

fa3e0a7

1 Parent(s): 8da2a5c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ from fastT5 import export_and_get_onnx_model
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 T5_tokenizer = AutoTokenizer.from_pretrained("jaimin/T5-Large")
-T5_model = export_and_get_onnx_model('jaimin/T5-large')

 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 T5_tokenizer = AutoTokenizer.from_pretrained("jaimin/T5-Large")
+T5_model = export_and_get_onnx_model('jaimin/T5-Large')