rishisriv-bh commited on
Commit
e967889
·
verified ·
1 Parent(s): da2c264

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,7 +1,8 @@
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
3
 
4
- client = InferenceClient("Qwen/Qwen1.5-32B-Chat")
 
5
 
6
  def build_prompt(system_message, history, user_message):
7
  prompt = f"<|im_start|>system\n{system_message}<|im_end|>\n"
 
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
3
 
4
+ hf_token = os.environ.get("HF_TOKEN")
5
+ client = InferenceClient("Qwen/Qwen1.5-32B-Chat", token=hf_token)
6
 
7
  def build_prompt(system_message, history, user_message):
8
  prompt = f"<|im_start|>system\n{system_message}<|im_end|>\n"