ysharma HF staff commited on
Commit
93e28b5
·
1 Parent(s): 0ba02be
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,7 +57,7 @@ def text_generate(prompt):
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[0]['generated_text'].split("\nQ:")[0] # +"."
61
  print(f"Final response after splits is: {solution}")
62
  #if "\n\n" not in output_tmp:
63
  # if output_tmp.find('.') != -1:
 
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: