Spaces:

IDEA-CCNL
/

Ziya-v1

Runtime error

Zimix commited on Jun 2, 2023

Commit

64be37c

1 Parent(s): 8e684e2

Update interaction.py

Files changed (1) hide show

interaction.py CHANGED Viewed

@@ -8,7 +8,7 @@ import gradio as gr
 from transformers import AutoTokenizer, LlamaForCausalLM
 from utils import SteamGenerationMixin
-auth_token = os.getenv("SECRET_TOKEN")
 class MindBot(object):
     def __init__(self, model_path, tokenizer_path,if_int8=False):

 from transformers import AutoTokenizer, LlamaForCausalLM
 from utils import SteamGenerationMixin
+auth_token = os.getenv("AUTH_TOKEN")
 class MindBot(object):
     def __init__(self, model_path, tokenizer_path,if_int8=False):