rliu commited on
Commit
0202baa
1 Parent(s): d7776c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -285,7 +285,6 @@ def preprocess_image(models, input_im, preprocess):
285
  start_time = time.time()
286
 
287
  if preprocess:
288
- input_im.thumbnail([1536, 1536], Image.Resampling.LANCZOS)
289
  input_im = load_and_preprocess(models['carvekit'], input_im)
290
  input_im = (input_im / 255.0).astype(np.float32)
291
  # (H, W, 3) array in [0, 1].
@@ -322,6 +321,7 @@ def main_run(models, device, cam_vis, return_what,
322
  :param raw_im (PIL Image).
323
  '''
324
 
 
325
  safety_checker_input = models['clip_fe'](raw_im, return_tensors='pt').to(device)
326
  (image, has_nsfw_concept) = models['nsfw'](
327
  images=np.ones((1, 3)), clip_input=safety_checker_input.pixel_values)
 
285
  start_time = time.time()
286
 
287
  if preprocess:
 
288
  input_im = load_and_preprocess(models['carvekit'], input_im)
289
  input_im = (input_im / 255.0).astype(np.float32)
290
  # (H, W, 3) array in [0, 1].
 
321
  :param raw_im (PIL Image).
322
  '''
323
 
324
+ raw_im.thumbnail([1536, 1536], Image.Resampling.LANCZOS)
325
  safety_checker_input = models['clip_fe'](raw_im, return_tensors='pt').to(device)
326
  (image, has_nsfw_concept) = models['nsfw'](
327
  images=np.ones((1, 3)), clip_input=safety_checker_input.pixel_values)