ndwdgda commited on
Commit
fec3d6a
·
verified ·
1 Parent(s): 3fdd1d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -5
app.py CHANGED
@@ -1,13 +1,17 @@
 
 
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,
12
  history: list[tuple[str, str]],
13
  system_message,
 
1
+
2
+ !pip install huggingfac_hub
3
  import gradio as gr
4
  from huggingface_hub import InferenceClient
5
 
6
  """
7
+ # Use a pipeline as a high-level helper
8
+ from transformers import pipeline
 
 
9
 
10
+ messages = [
11
+ {"role": "user", "content": "Who are you?"},
12
+ ]
13
+ pipe = pipeline("text-generation", model="ibm-granite/granite-8b-code-instruct")
14
+ pipe(messages)
15
  message,
16
  history: list[tuple[str, str]],
17
  system_message,