Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,10 +1,16 @@
|
|
1 |
import gradio as gr
|
|
|
2 |
from huggingface_hub import InferenceClient
|
3 |
|
4 |
-
""
|
5 |
-
|
6 |
-
""
|
7 |
-
|
|
|
|
|
|
|
|
|
|
|
8 |
|
9 |
|
10 |
def respond(
|
|
|
1 |
import gradio as gr
|
2 |
+
import os
|
3 |
from huggingface_hub import InferenceClient
|
4 |
|
5 |
+
HF_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
|
6 |
+
|
7 |
+
client = InferenceClient(model="HuggingFaceH4/zephyr-7b-beta", token=HF_TOKEN)
|
8 |
+
|
9 |
+
|
10 |
+
def respond(message):
|
11 |
+
for response in client.chat_completion(messages=[{"role": "user", "content": message}]):
|
12 |
+
yield response["choices"][0]["message"]["content"]
|
13 |
+
|
14 |
|
15 |
|
16 |
def respond(
|