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 |
|