Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -60,7 +60,7 @@ def gradio(model, tokenizer):
|
|
60 |
input_ids,
|
61 |
pad_token_id=tokenizer.pad_token_id,
|
62 |
max_new_tokens=255,
|
63 |
-
|
64 |
top_k=top_k,
|
65 |
top_p=top_p,
|
66 |
no_repeat_ngram_size=no_repeat_ngram_size,
|
@@ -79,7 +79,7 @@ def gradio(model, tokenizer):
|
|
79 |
|
80 |
return text
|
81 |
|
82 |
-
bot = gr.Chatbot(
|
83 |
|
84 |
with gr.Blocks() as demo:
|
85 |
gr.Markdown("GPT2 chatbot | Powered by nlp-greyfoss")
|
|
|
60 |
input_ids,
|
61 |
pad_token_id=tokenizer.pad_token_id,
|
62 |
max_new_tokens=255,
|
63 |
+
num_beams=1, # with cpu
|
64 |
top_k=top_k,
|
65 |
top_p=top_p,
|
66 |
no_repeat_ngram_size=no_repeat_ngram_size,
|
|
|
79 |
|
80 |
return text
|
81 |
|
82 |
+
bot = gr.Chatbot(show_copy_button=True, show_share_button=True, scroll_to_output=True, height="1000")
|
83 |
|
84 |
with gr.Blocks() as demo:
|
85 |
gr.Markdown("GPT2 chatbot | Powered by nlp-greyfoss")
|