cocktailpeanut commited on
Commit
007c79c
1 Parent(s): 711724d
Files changed (1) hide show
  1. app.py +18 -18
app.py CHANGED
@@ -10,7 +10,7 @@ from diffusers import StableCascadeDecoderPipeline, StableCascadePriorPipeline
10
  from diffusers.pipelines.wuerstchen import DEFAULT_STAGE_C_TIMESTEPS
11
  #import spaces
12
  from previewer.modules import Previewer
13
- import user_history
14
 
15
  os.environ['TOKENIZERS_PARALLELISM'] = 'false'
16
 
@@ -128,23 +128,23 @@ def generate(
128
  output_type="pil",
129
  ).images
130
 
131
- #Save images
132
- for image in decoder_output:
133
- user_history.save_image(
134
- profile=profile,
135
- image=image,
136
- label=prompt,
137
- metadata={
138
- "negative_prompt": negative_prompt,
139
- "seed": seed,
140
- "width": width,
141
- "height": height,
142
- "prior_guidance_scale": prior_guidance_scale,
143
- "decoder_num_inference_steps": decoder_num_inference_steps,
144
- "decoder_guidance_scale": decoder_guidance_scale,
145
- "num_images_per_prompt": num_images_per_prompt,
146
- },
147
- )
148
 
149
  yield decoder_output[0]
150
 
 
10
  from diffusers.pipelines.wuerstchen import DEFAULT_STAGE_C_TIMESTEPS
11
  #import spaces
12
  from previewer.modules import Previewer
13
+ #import user_history
14
 
15
  os.environ['TOKENIZERS_PARALLELISM'] = 'false'
16
 
 
128
  output_type="pil",
129
  ).images
130
 
131
+ # #Save images
132
+ # for image in decoder_output:
133
+ # user_history.save_image(
134
+ # profile=profile,
135
+ # image=image,
136
+ # label=prompt,
137
+ # metadata={
138
+ # "negative_prompt": negative_prompt,
139
+ # "seed": seed,
140
+ # "width": width,
141
+ # "height": height,
142
+ # "prior_guidance_scale": prior_guidance_scale,
143
+ # "decoder_num_inference_steps": decoder_num_inference_steps,
144
+ # "decoder_guidance_scale": decoder_guidance_scale,
145
+ # "num_images_per_prompt": num_images_per_prompt,
146
+ # },
147
+ # )
148
 
149
  yield decoder_output[0]
150