baptiste.bernard commited on
Commit
9cc30f1
·
1 Parent(s): 269ff46

update access token

Browse files
Files changed (2) hide show
  1. app.py +9 -4
  2. requirements.txt +1 -0
app.py CHANGED
@@ -1,8 +1,13 @@
 
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
3
  import chardet
4
 
5
- client = InferenceClient("HuggingFaceH4/zephyr-7b-beta", token="") #generate Access tokens
 
 
 
 
6
 
7
  file_content = None
8
 
@@ -15,14 +20,14 @@ def respond(message, history, system_message, max_tokens, temperature, top_p, fi
15
  messages.append({"role": "user", "content": val[0]})
16
  if val[1]:
17
  messages.append({"role": "assistant", "content": val[1]})
18
-
19
  messages.append({"role": "user", "content": message})
20
 
21
  response = ""
22
 
23
  if file:
24
  try:
25
- file_content = file.decode("utf-8")
26
  except UnicodeDecodeError:
27
  result = chardet.detect(file)
28
  encoding = result['encoding']
@@ -53,7 +58,7 @@ with gr.Blocks() as demo:
53
  gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
54
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
55
  gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
56
- gr.File(label="Télécharger un fichier", type="binary"),
57
  ],
58
  )
59
 
 
1
+ import os
2
  import gradio as gr
3
  from huggingface_hub import InferenceClient
4
  import chardet
5
 
6
+ # Récupérez le token à partir de la variable d'environnement
7
+ token = os.getenv("HUGGINGFACE_TOKEN")
8
+
9
+ # Utilisez le token pour initialiser le client
10
+ client = InferenceClient("HuggingFaceH4/zephyr-7b-beta", token=token)
11
 
12
  file_content = None
13
 
 
20
  messages.append({"role": "user", "content": val[0]})
21
  if val[1]:
22
  messages.append({"role": "assistant", "content": val[1]})
23
+
24
  messages.append({"role": "user", "content": message})
25
 
26
  response = ""
27
 
28
  if file:
29
  try:
30
+ file_content = file.decode("utf-8")
31
  except UnicodeDecodeError:
32
  result = chardet.detect(file)
33
  encoding = result['encoding']
 
58
  gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
59
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
60
  gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
61
+ gr.File(label="Télécharger un fichier", type="binary"),
62
  ],
63
  )
64
 
requirements.txt CHANGED
@@ -1,2 +1,3 @@
1
  chardet
2
  huggingface_hub==0.25.2
 
 
1
  chardet
2
  huggingface_hub==0.25.2
3
+ os