Update app.py
Browse files
app.py
CHANGED
@@ -98,6 +98,9 @@ def generate_image(enhanced_caption, seed, num_inference_steps, guidance_scale):
|
|
98 |
generator=generator,
|
99 |
system_prompt="You are an assistant designed to generate superior images with the superior degree of image-text alignment based on textual prompts or user prompts.",
|
100 |
).images[0]
|
|
|
|
|
|
|
101 |
pipe.to("cpu")
|
102 |
torch.cuda.empty_cache()
|
103 |
|
|
|
98 |
generator=generator,
|
99 |
system_prompt="You are an assistant designed to generate superior images with the superior degree of image-text alignment based on textual prompts or user prompts.",
|
100 |
).images[0]
|
101 |
+
|
102 |
+
print(image)
|
103 |
+
|
104 |
pipe.to("cpu")
|
105 |
torch.cuda.empty_cache()
|
106 |
|