stzhao commited on
Commit
c5b92f1
·
verified ·
1 Parent(s): 7a858c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -86,6 +86,7 @@ def generate_image(enhanced_caption, seed, num_inference_steps, guidance_scale):
86
 
87
  return image
88
 
 
89
  def run_pipeline(image_caption, text_caption, seed, num_inference_steps, guidance_scale):
90
  """Run the complete pipeline from captions to final image"""
91
  combined_caption, enhanced_caption = generate_enhanced_caption(image_caption, text_caption)
 
86
 
87
  return image
88
 
89
+ @spaces.GPU(duration=100)
90
  def run_pipeline(image_caption, text_caption, seed, num_inference_steps, guidance_scale):
91
  """Run the complete pipeline from captions to final image"""
92
  combined_caption, enhanced_caption = generate_enhanced_caption(image_caption, text_caption)