ysharma HF staff commited on
Commit
650be9a
·
1 Parent(s): c85a2a5
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -81,8 +81,13 @@ def text_generate(prompt, generated_txt): #, input_prompt_sql ): #, input_prompt
81
  # poem = output_tmp.split("\n\n")[0] # +"."
82
  #poem = poem.replace('?','')
83
  #print(f"Poem being returned is: {poem}")
84
- prompt = final_solution[len(prompt):]
85
- return generated_txt+prompt, prompt #final_solution
 
 
 
 
 
86
 
87
 
88
  demo = gr.Blocks()
 
81
  # poem = output_tmp.split("\n\n")[0] # +"."
82
  #poem = poem.replace('?','')
83
  #print(f"Poem being returned is: {poem}")
84
+ if len(generated_txt) == 0 :
85
+ display_output = final_solution
86
+ else:
87
+ display_output = generated_txt[:len(prompt)] + final_solution
88
+ new_prompt = final_solution[len(prompt):]
89
+
90
+ return display_output, new_prompt #generated_txt+prompt #final_solution
91
 
92
 
93
  demo = gr.Blocks()