Humphreykowl commited on
Commit
bae280b
·
verified ·
1 Parent(s): 405e531

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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(["简约", "复古", "现代", "街头"])