ochyai commited on
Commit
290fc46
1 Parent(s): 1bc29e0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -78,10 +78,10 @@ class OpenAI:
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
 
@@ -90,7 +90,7 @@ class OpenAI:
90
  #else:
91
  # visualize_prompt = "vacant dish"
92
 
93
- #print("visualize_prompt:"+visualize_prompt)
94
 
95
  answers = stability_api.generate(
96
  prompt=visualize_prompt,
 
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 = "vacant dish" # or any default value
85
 
86
  #print("split_content:"+split_content)
87
 
 
90
  #else:
91
  # visualize_prompt = "vacant dish"
92
 
93
+ print("visualize_prompt:"+visualize_prompt)
94
 
95
  answers = stability_api.generate(
96
  prompt=visualize_prompt,