Spaces:

yxccai
/

text-style-api

Sleeping

yxccai commited on May 31

Commit

7fab2f7

verified ·

1 Parent(s): ae39785

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ gc.collect()
 os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:128"
 # 模型名称
-model_name = "您的用户名/text-style-converter"
 # 全局变量存储模型
 tokenizer = None

 os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:128"
 # 模型名称
+model_name = "yxccai/text-style-converter"
 # 全局变量存储模型
 tokenizer = None