Spaces:

VietCat
/

gpt2-vietnamese-api

Running

VietCat commited on May 8

Commit

535d128

1 Parent(s): e3ea79a

adjust generation time

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import time
 import warnings
 import re
 import gc
-warnings.filterWarnings("ignore", category=UserWarning, module="torch._utils")
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 import torch

 import warnings
 import re
 import gc
+warnings.filterwarnings("ignore", category=UserWarning, module="torch._utils")
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 import torch