Saif Rehman Nasir commited on
Commit
1a0060e
·
1 Parent(s): 78fc2e0
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,7 +15,7 @@ def generate_text(context, num_of_tokens, temperature=1.0):
15
  return decode(model.generate(idx, max_new_tokens=num_of_tokens,temperature=temperature)[0].tolist())
16
 
17
 
18
- with gr.Blocks as demo:
19
  gr.HTML("<h1 align='center'> Shakespeare Text Generator</h1>")
20
 
21
  context = gr.Textbox(label = "Enter context (optional)")
@@ -29,7 +29,7 @@ with gr.Blocks as demo:
29
  num_of_tokens,tmp
30
  ]
31
  generate_btn = gr.Button(value="Generate")
32
- outputs = [gr.Textbox(label= "Generated text")]
33
  generate_btn.click(fn = generate_text, inputs= inputs, outputs= outputs)
34
 
35
  demo.launch()
 
15
  return decode(model.generate(idx, max_new_tokens=num_of_tokens,temperature=temperature)[0].tolist())
16
 
17
 
18
+ with gr.Blocks() as demo:
19
  gr.HTML("<h1 align='center'> Shakespeare Text Generator</h1>")
20
 
21
  context = gr.Textbox(label = "Enter context (optional)")
 
29
  num_of_tokens,tmp
30
  ]
31
  generate_btn = gr.Button(value="Generate")
32
+ outputs = [gr.Textbox(label= "Generated text: ")]
33
  generate_btn.click(fn = generate_text, inputs= inputs, outputs= outputs)
34
 
35
  demo.launch()