Spaces:
Running
Running
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
|