ochyai commited on
Commit
1bc29e0
1 Parent(s): 1c36440

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -77,7 +77,11 @@ class OpenAI:
77
 
78
  content = result["choices"][0]["message"]["content"].strip()
79
 
80
- visualize_prompt = content.split("### Prompt for Visual Expression\n\n")[1]
 
 
 
 
81
 
82
  #print("split_content:"+split_content)
83
 
 
77
 
78
  content = result["choices"][0]["message"]["content"].strip()
79
 
80
+ split_content = content.split("### Prompt for Visual Expression\n\n")
81
+ if len(split_content) > 1:
82
+ visualize_prompt = split_content[1]
83
+ else:
84
+ visualize_prompt = None # or any default value
85
 
86
  #print("split_content:"+split_content)
87