Stable-X commited on
Commit
6a048d4
1 Parent(s): 5f27781

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -86,7 +86,7 @@ def resize_image(input_image, resolution):
86
  W = int(np.round(W / 64.0)) * 64
87
 
88
  # Resize the image using PIL's resize method
89
- img = input_image.resize((W, H), Image.LANCZOS if k > 1 else Image.ANTIALIAS)
90
 
91
  return img
92
 
@@ -99,7 +99,7 @@ def process_image(
99
 
100
  path_output_dir = tempfile.mkdtemp()
101
  path_out_png = os.path.join(path_output_dir, f"{name_base}_normal_colored.png")
102
-
103
  input_image = Image.open(path_input)
104
  input_image = resize_image(input_image, default_image_processing_resolution)
105
 
@@ -112,7 +112,7 @@ def process_image(
112
  normal_pred = pipe_out.prediction[0, :, :]
113
  normal_colored = pipe.image_processor.visualize_normals(pipe_out.prediction)
114
  normal_colored[-1].save(path_out_png)
115
- return [input_image, path_out_png]
116
 
117
  def center_crop(img):
118
  # Open the image file
 
86
  W = int(np.round(W / 64.0)) * 64
87
 
88
  # Resize the image using PIL's resize method
89
+ img = input_image.resize((W, H), Image.LANCZOS)
90
 
91
  return img
92
 
 
99
 
100
  path_output_dir = tempfile.mkdtemp()
101
  path_out_png = os.path.join(path_output_dir, f"{name_base}_normal_colored.png")
102
+ yield None
103
  input_image = Image.open(path_input)
104
  input_image = resize_image(input_image, default_image_processing_resolution)
105
 
 
112
  normal_pred = pipe_out.prediction[0, :, :]
113
  normal_colored = pipe.image_processor.visualize_normals(pipe_out.prediction)
114
  normal_colored[-1].save(path_out_png)
115
+ yield [input_image, path_out_png]
116
 
117
  def center_crop(img):
118
  # Open the image file