serhii-korobchenko commited on
Commit
e92f3f5
1 Parent(s): 0fd72b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def complete_text(start_of_sentence):
49
  result_clear = model_clear.generate(start_of_sentence, max_length=100)
50
  result = model.generate(start_of_sentence, max_length=100)
51
  #result = model(start_of_sentence, max_length=50, do_sample=True)[0]['generated_text']
52
- return result
53
 
54
  model_clear, model = download_model_NLP()
55
  iface = gr.Interface(fn=complete_text, inputs=gr.inputs.Textbox(lines=2, placeholder='Start of Sentence Here...'), outputs=["text", "text"],)
 
49
  result_clear = model_clear.generate(start_of_sentence, max_length=100)
50
  result = model.generate(start_of_sentence, max_length=100)
51
  #result = model(start_of_sentence, max_length=50, do_sample=True)[0]['generated_text']
52
+ return result_clear, result
53
 
54
  model_clear, model = download_model_NLP()
55
  iface = gr.Interface(fn=complete_text, inputs=gr.inputs.Textbox(lines=2, placeholder='Start of Sentence Here...'), outputs=["text", "text"],)