stzhao commited on
Commit
e435d1a
·
verified ·
1 Parent(s): a826a23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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