Martial667 commited on
Commit
5f93b0b
·
verified ·
1 Parent(s): 76caddd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +22 -30
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
3
 
 
4
  def respond(
5
  message,
6
  history: list[dict[str, str]],
@@ -10,13 +11,15 @@ def respond(
10
  top_p,
11
  hf_token: gr.OAuthToken,
12
  ):
13
- if hf_token is None:
14
- raise gr.Error("Veuillez vous connecter via le bouton pour utiliser le chatbot.")
15
-
16
  client = InferenceClient(token=hf_token.token, model="deepseek-ai/DeepSeek-V3.2")
17
 
18
  messages = [{"role": "system", "content": system_message}]
 
19
  messages.extend(history)
 
20
  messages.append({"role": "user", "content": message})
21
 
22
  response = ""
@@ -36,43 +39,32 @@ def respond(
36
  response += token
37
  yield response
38
 
39
- theme = gr.themes.Soft(
40
- primary_hue="indigo",
41
- secondary_hue="violet",
42
- )
43
 
44
- css = """
45
- h1 { text-align: center; color: #4f46e5; }
46
  """
47
-
48
- my_chatbot = gr.Chatbot(
49
- height=600,
50
- type="messages",
51
- avatar_images=(None, "https://huggingface.co/datasets/huggingface/brand-assets/resolve/main/hf-logo.svg"),
52
- show_copy_button=True
53
- )
54
-
55
- chat = gr.ChatInterface(
56
  respond,
57
- chatbot=my_chatbot,
58
  type="messages",
59
- title="Le chatbot qu'il vous faut",
60
  additional_inputs=[
61
  gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
62
  gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
63
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
64
- gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p"),
 
 
 
 
 
 
65
  ],
66
  )
67
 
68
- with gr.Blocks(theme=theme, css=css) as demo:
69
- with gr.Row():
70
- with gr.Column(scale=1, min_width=200):
71
- gr.Markdown("### Authentification")
72
- gr.LoginButton()
73
-
74
- with gr.Column(scale=4):
75
- chat.render()
76
 
77
  if __name__ == "__main__":
78
- demo.launch(ssr_mode=False)
 
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
3
 
4
+
5
  def respond(
6
  message,
7
  history: list[dict[str, str]],
 
11
  top_p,
12
  hf_token: gr.OAuthToken,
13
  ):
14
+ """
15
+ For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
16
+ """
17
  client = InferenceClient(token=hf_token.token, model="deepseek-ai/DeepSeek-V3.2")
18
 
19
  messages = [{"role": "system", "content": system_message}]
20
+
21
  messages.extend(history)
22
+
23
  messages.append({"role": "user", "content": message})
24
 
25
  response = ""
 
39
  response += token
40
  yield response
41
 
 
 
 
 
42
 
 
 
43
  """
44
+ For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
45
+ """
46
+ chatbot = gr.ChatInterface(
 
 
 
 
 
 
47
  respond,
 
48
  type="messages",
 
49
  additional_inputs=[
50
  gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
51
  gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
52
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
53
+ gr.Slider(
54
+ minimum=0.1,
55
+ maximum=1.0,
56
+ value=0.95,
57
+ step=0.05,
58
+ label="Top-p (nucleus sampling)",
59
+ ),
60
  ],
61
  )
62
 
63
+ with gr.Blocks() as demo:
64
+ with gr.Sidebar():
65
+ gr.LoginButton()
66
+ chatbot.render()
67
+
 
 
 
68
 
69
  if __name__ == "__main__":
70
+ demo.launch(share=True)