Update app.py
Browse files
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)
|