SIUBIU commited on
Commit
4b81bf3
·
verified ·
1 Parent(s): 43cea68

Update clothGen.py

Browse files
Files changed (1) hide show
  1. clothGen.py +1 -1
clothGen.py CHANGED
@@ -24,7 +24,7 @@ def cloth_gen(advice, gender):
24
  promptGen = ""
25
  if start_index != -1:
26
  start_index += len("prompt\n")
27
- end_index = prompt.find("promptEnd")
28
  if end_index != -1:
29
  extracted_content = prompt[start_index:end_index]
30
  promptGen = extracted_content
 
24
  promptGen = ""
25
  if start_index != -1:
26
  start_index += len("prompt\n")
27
+ end_index = prompt.find("End" or "end")
28
  if end_index != -1:
29
  extracted_content = prompt[start_index:end_index]
30
  promptGen = extracted_content