Martin van Dijk commited on
Commit
abf3d23
·
1 Parent(s): af6f5f2

Update space

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -1,11 +1,10 @@
1
  import gradio as gr
2
- from transformers import pipeline
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
 
8
- model = pipeline("text-generation")
9
 
10
 
11
  def predict(prompt):
 
1
  import gradio as gr
2
+ from huggingface_hub import InferenceClient
 
3
  """
4
  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
5
  """
6
 
7
+ model = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
8
 
9
 
10
  def predict(prompt):