Spaces:

yxccai
/

text-style-api

Running

yxccai commited on 26 days ago

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