Meaowangxi commited on
Commit
8755d5a
·
verified ·
1 Parent(s): ac12192

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -100,12 +100,13 @@ def task1_StyleTransfer(photo, blur_radius, sketch):
100
  enhancer = ImageEnhance.Contrast(processed_image)
101
  processed_image = enhancer.enhance(contrast_factor)
102
 
103
- images = ip_model.generate(pil_image=photoImage, image=processed_image, num_samples=1, num_inference_steps=30, seed=52)
104
- original = image_grid(images, 1, 1)
105
- images = ip_model.generate(pil_image=blurPhoto, image=processed_image, num_samples=1, num_inference_steps=30, seed=52)
106
  result= image_grid(images, 1, 1)
107
 
108
- return original,result
 
109
 
110
  def task1_test(photo, blur_radius, sketch):
111
  original = photo
 
100
  enhancer = ImageEnhance.Contrast(processed_image)
101
  processed_image = enhancer.enhance(contrast_factor)
102
 
103
+ #images = ip_model.generate(pil_image=photoImage, image=processed_image, num_samples=1, num_inference_steps=15, seed=52)
104
+ #original = image_grid(images, 1, 1)
105
+ images = ip_model.generate(pil_image=blurPhoto, image=processed_image, num_samples=1, num_inference_steps=10, seed=52)
106
  result= image_grid(images, 1, 1)
107
 
108
+ #return original,result
109
+ return blurPhoto, result
110
 
111
  def task1_test(photo, blur_radius, sketch):
112
  original = photo