shellypeng commited on
Commit
ca20cf0
·
verified ·
1 Parent(s): b01de6c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -219,10 +219,10 @@ def real_img2img_to_anime(text, neg_prompt_box, input_img):
219
  negative_prompt = neg_prompt_box + hidden_negative
220
  negative_prompt_embeds = compel_proc(negative_prompt)
221
  # input_img = depth_estimator(input_img)['depth']
222
- res_image0 = pipe_img2img(image=input_img, strength=0.6, prompt_embeds=prompt_embeds, negative_prompt_embeds=negative_prompt_embeds, num_inference_steps=40).images[0]
223
- res_image1 = pipe_img2img(image=input_img, strength=0.6, prompt_embeds=prompt_embeds, negative_prompt_embeds=negative_prompt_embeds, num_inference_steps=40).images[0]
224
- res_image2 = pipe_img2img(image=input_img, strength=0.6, prompt_embeds=prompt_embeds, negative_prompt_embeds=negative_prompt_embeds, num_inference_steps=40).images[0]
225
- res_image3 = pipe_img2img(image=input_img, strength=0.6, prompt_embeds=prompt_embeds, negative_prompt_embeds=negative_prompt_embeds, num_inference_steps=40).images[0]
226
 
227
  return res_image0, res_image1, res_image2, res_image3
228
 
 
219
  negative_prompt = neg_prompt_box + hidden_negative
220
  negative_prompt_embeds = compel_proc(negative_prompt)
221
  # input_img = depth_estimator(input_img)['depth']
222
+ res_image0 = pipe_img2img(image=input_img, strength=0.8, prompt_embeds=prompt_embeds, negative_prompt_embeds=negative_prompt_embeds, num_inference_steps=40).images[0]
223
+ res_image1 = pipe_img2img(image=input_img, strength=0.8, prompt_embeds=prompt_embeds, negative_prompt_embeds=negative_prompt_embeds, num_inference_steps=40).images[0]
224
+ res_image2 = pipe_img2img(image=input_img, strength=0.8, prompt_embeds=prompt_embeds, negative_prompt_embeds=negative_prompt_embeds, num_inference_steps=40).images[0]
225
+ res_image3 = pipe_img2img(image=input_img, strength=0.8, prompt_embeds=prompt_embeds, negative_prompt_embeds=negative_prompt_embeds, num_inference_steps=40).images[0]
226
 
227
  return res_image0, res_image1, res_image2, res_image3
228