DataBob commited on
Commit
f40151d
·
1 Parent(s): b7cbe1b
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -1,11 +1,15 @@
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
 
 
 
3
 
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(
11
  message,
@@ -25,6 +29,7 @@ def respond(
25
 
26
  messages.append({"role": "user", "content": message})
27
 
 
28
  response = ""
29
 
30
  for message in client.chat_completion(
 
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
3
+ import logging
4
+
5
+ logger = logging.getLogger(__name__)
6
 
7
  """
8
  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
9
  """
10
  client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
11
 
12
+ logger.info("### Inference client: zephyr-7b-beta")
13
 
14
  def respond(
15
  message,
 
29
 
30
  messages.append({"role": "user", "content": message})
31
 
32
+ logger.info(messages)
33
  response = ""
34
 
35
  for message in client.chat_completion(