ajeetkumar01 commited on
Commit
ed9a903
·
verified ·
1 Parent(s): 09b124b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -54,13 +54,13 @@ def generate_text_with_nucleus_search(input_text, max_length=16, do_sample=True,
54
  output = model.generate(input_ids, max_length=max_length, do_sample=do_sample, top_p=top_p)
55
  # Decode the generated output
56
  generated_text = tokenizer.decode(output[0])
57
- return generate_text_with_nucleus_search
58
 
59
 
60
  # Create Gradio interface
61
  input_text = gr.Textbox(lines=10, label="Input Text", placeholder="Enter text for text generation...")
62
  output_text1 = gr.Textbox(label="Generated Text")
63
- output_text2 = gr.Textbox(lable="generate_text_with_nucleus_search")
64
 
65
 
66
  gr.Interface(generate_text, input_text, output_text1,output_text2,
 
54
  output = model.generate(input_ids, max_length=max_length, do_sample=do_sample, top_p=top_p)
55
  # Decode the generated output
56
  generated_text = tokenizer.decode(output[0])
57
+ return generated_text
58
 
59
 
60
  # Create Gradio interface
61
  input_text = gr.Textbox(lines=10, label="Input Text", placeholder="Enter text for text generation...")
62
  output_text1 = gr.Textbox(label="Generated Text")
63
+ output_text2 = gr.Textbox(label="generate_text_with_nucleus_search")
64
 
65
 
66
  gr.Interface(generate_text, input_text, output_text1,output_text2,