Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,7 +24,7 @@ load_model("LeCarnet-8M")
|
|
| 24 |
# Streaming generation function
|
| 25 |
def respond(message, max_tokens, temperature, top_p):
|
| 26 |
inputs = tokenizer(message, return_tensors="pt")
|
| 27 |
-
streamer = TextIteratorStreamer(tokenizer, skip_prompt=
|
| 28 |
|
| 29 |
generate_kwargs = dict(
|
| 30 |
**inputs,
|
|
@@ -66,7 +66,7 @@ def update_model(model_name):
|
|
| 66 |
with gr.Blocks(title="LeCarnet - Chat Interface") as demo:
|
| 67 |
with gr.Row(equal_height=True):
|
| 68 |
gr.Markdown(
|
| 69 |
-
f'<img src="
|
| 70 |
elem_classes="header-image"
|
| 71 |
)
|
| 72 |
# Title next to image
|
|
|
|
| 24 |
# Streaming generation function
|
| 25 |
def respond(message, max_tokens, temperature, top_p):
|
| 26 |
inputs = tokenizer(message, return_tensors="pt")
|
| 27 |
+
streamer = TextIteratorStreamer(tokenizer, skip_prompt=False, skip_special_tokens=True)
|
| 28 |
|
| 29 |
generate_kwargs = dict(
|
| 30 |
**inputs,
|
|
|
|
| 66 |
with gr.Blocks(title="LeCarnet - Chat Interface") as demo:
|
| 67 |
with gr.Row(equal_height=True):
|
| 68 |
gr.Markdown(
|
| 69 |
+
f'<img src="media/le-carnet.png" style="height:80px; width:80px; margin-right:10px; vertical-align:top;">',
|
| 70 |
elem_classes="header-image"
|
| 71 |
)
|
| 72 |
# Title next to image
|