xyizko commited on
Commit
87569bb
·
verified ·
1 Parent(s): 1201506

Update app.py

Browse files

- Add HF Token and Model Entry

Files changed (1) hide show
  1. app.py +17 -1
app.py CHANGED
@@ -4,7 +4,19 @@ from huggingface_hub import InferenceClient
4
  """
5
  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
6
  """
7
- client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
 
 
 
 
 
 
 
 
 
 
 
8
 
9
 
10
  def respond(
@@ -40,12 +52,16 @@ def respond(
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
  demo = gr.ChatInterface(
47
  respond,
48
  additional_inputs=[
 
 
49
  gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
50
  gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
51
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
 
4
  """
5
  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
6
  """
7
+
8
+ # Getting the HF Token and Well as the model
9
+ token_input = gr.Textbox(type="password", label="HF API Token")
10
+
11
+ if not token:
12
+ return "Please provide an HF API Token."
13
+
14
+ model_input = gr.Textbox(label="Model Name", value="HuggingFaceH4/zephyr-7b-beta")
15
+
16
+ if not model_input:
17
+ model_input = "meta-llama/Llama-3.1-8B-Instruct"
18
+
19
+ client = InferenceClient(model_input)
20
 
21
 
22
  def respond(
 
52
  yield response
53
 
54
 
55
+
56
+
57
  """
58
  For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
59
  """
60
  demo = gr.ChatInterface(
61
  respond,
62
  additional_inputs=[
63
+ token_input,
64
+ model_input,
65
  gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
66
  gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
67
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),