minnehwg commited on
Commit
460c4fc
·
verified ·
1 Parent(s): 9a3fa77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ tokenizer, model = get_model()
21
  # )
22
 
23
  def generate(text):
24
- return pipeline(text, )
25
 
26
  with gr.Blocks() as demo:
27
  gr.Markdown("Start typing below and then click **Run** to see the output.")
 
21
  # )
22
 
23
  def generate(text):
24
+ return pipeline(text, model, tokenizer)
25
 
26
  with gr.Blocks() as demo:
27
  gr.Markdown("Start typing below and then click **Run** to see the output.")