Spaces:
Running
on
Zero
Running
on
Zero
UI improvement (#1)
Browse files- Improve UI (4e3e7a774037eef91faddd58db3b1a4b55f5baf6)
Co-authored-by: hayas <[email protected]>
README.md
CHANGED
|
@@ -4,7 +4,7 @@ emoji: 💬
|
|
| 4 |
colorFrom: yellow
|
| 5 |
colorTo: purple
|
| 6 |
sdk: gradio
|
| 7 |
-
sdk_version: 5.0
|
| 8 |
app_file: app.py
|
| 9 |
pinned: false
|
| 10 |
license: mit
|
|
|
|
| 4 |
colorFrom: yellow
|
| 5 |
colorTo: purple
|
| 6 |
sdk: gradio
|
| 7 |
+
sdk_version: 5.20.0
|
| 8 |
app_file: app.py
|
| 9 |
pinned: false
|
| 10 |
license: mit
|
app.py
CHANGED
|
@@ -1,7 +1,6 @@
|
|
| 1 |
# Ref: https://huggingface.co/spaces/ysharma/Chat_with_Meta_llama3_8b
|
| 2 |
|
| 3 |
import gradio as gr
|
| 4 |
-
import os
|
| 5 |
import spaces
|
| 6 |
import torch
|
| 7 |
from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
|
|
@@ -92,12 +91,11 @@ def chat_llama3_8b(message: str,
|
|
| 92 |
outputs = []
|
| 93 |
for text in streamer:
|
| 94 |
outputs.append(text)
|
| 95 |
-
print(outputs)
|
| 96 |
yield "".join(outputs)
|
| 97 |
|
| 98 |
|
| 99 |
# Gradio block
|
| 100 |
-
chatbot=gr.Chatbot(height=450, placeholder=PLACEHOLDER, label='Gradio ChatInterface')
|
| 101 |
|
| 102 |
with gr.Blocks(fill_height=True, css=css) as demo:
|
| 103 |
|
|
|
|
| 1 |
# Ref: https://huggingface.co/spaces/ysharma/Chat_with_Meta_llama3_8b
|
| 2 |
|
| 3 |
import gradio as gr
|
|
|
|
| 4 |
import spaces
|
| 5 |
import torch
|
| 6 |
from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
|
|
|
|
| 91 |
outputs = []
|
| 92 |
for text in streamer:
|
| 93 |
outputs.append(text)
|
|
|
|
| 94 |
yield "".join(outputs)
|
| 95 |
|
| 96 |
|
| 97 |
# Gradio block
|
| 98 |
+
chatbot=gr.Chatbot(height=450, placeholder=PLACEHOLDER, label='Gradio ChatInterface', scale=1)
|
| 99 |
|
| 100 |
with gr.Blocks(fill_height=True, css=css) as demo:
|
| 101 |
|