Update app.py
Browse files
app.py
CHANGED
|
@@ -40,6 +40,8 @@ class SimpleModelManager:
|
|
| 40 |
logger.info("使用简化模型管理器")
|
| 41 |
|
| 42 |
def generate_caption(self, image):
|
|
|
|
|
|
|
| 43 |
return "一件时尚的{}服装,采用{}设计".format(
|
| 44 |
random.choice(["夏季", "冬季", "春秋季"]),
|
| 45 |
random.choice(["简约", "复古", "现代", "街头"])
|
|
|
|
| 40 |
logger.info("使用简化模型管理器")
|
| 41 |
|
| 42 |
def generate_caption(self, image):
|
| 43 |
+
if 'last_generated_design' not in globals() or last_generated_design is None:
|
| 44 |
+
return '请先生成建议设计'
|
| 45 |
return "一件时尚的{}服装,采用{}设计".format(
|
| 46 |
random.choice(["夏季", "冬季", "春秋季"]),
|
| 47 |
random.choice(["简约", "复古", "现代", "街头"])
|