ysharma HF staff commited on
Commit
f8ce710
·
1 Parent(s): 93e28b5
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -56,8 +56,8 @@ def text_generate(prompt):
56
  output = response.json()
57
  print(f"output is : {output}") #{output}")
58
  output_tmp = output[0]['generated_text']
59
- print(f"Bloom response without splits is: {output_tmp}")
60
- solution = output_tmp.split("Q:")[0] #output[0]['generated_text'].split("Q:")[0] # +"."
61
  print(f"Final response after splits is: {solution}")
62
  #if "\n\n" not in output_tmp:
63
  # if output_tmp.find('.') != -1:
 
56
  output = response.json()
57
  print(f"output is : {output}") #{output}")
58
  output_tmp = output[0]['generated_text']
59
+ print(f"output_tmp is: {output_tmp}")
60
+ solution = output_tmp.split("\nQ:")[0] #output[0]['generated_text'].split("Q:")[0] # +"."
61
  print(f"Final response after splits is: {solution}")
62
  #if "\n\n" not in output_tmp:
63
  # if output_tmp.find('.') != -1: