greyfoss commited on
Commit
ac50f75
·
verified ·
1 Parent(s): f1741c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- # num_beams=3,
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(scale=8)
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")