model2 commited on
Commit
211faaf
·
1 Parent(s): 4bb9b6b

image sizing

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -92,7 +92,7 @@ FACE_MODEL = reactorloadfacemodel.load_model(
92
  imageresize = NODE_CLASS_MAPPINGS["ImageResize+"]()
93
  reactorfaceswap = NODE_CLASS_MAPPINGS["ReActorFaceSwap"]()
94
  imageupscalewithmodel = NODE_CLASS_MAPPINGS["ImageUpscaleWithModel"]()
95
- saveimage = NODE_CLASS_MAPPINGS["SaveImage"]()
96
  UPSCALE_MODEL = upscalemodelloader.load_model(model_name="ESRGAN/4x_NMKD-Siax_200k.pth")
97
 
98
 
@@ -258,8 +258,9 @@ def advance_blur(input_image):
258
  image=get_value_at_index(upscaled_image, 0),
259
  )
260
 
261
- saved_image = saveimage.save_images(
262
- filename_prefix="advance_blur",
 
263
  images=get_value_at_index(final_image, 0),
264
  )
265
 
 
92
  imageresize = NODE_CLASS_MAPPINGS["ImageResize+"]()
93
  reactorfaceswap = NODE_CLASS_MAPPINGS["ReActorFaceSwap"]()
94
  imageupscalewithmodel = NODE_CLASS_MAPPINGS["ImageUpscaleWithModel"]()
95
+ local_save = NODE_CLASS_MAPPINGS["Local Save"]()
96
  UPSCALE_MODEL = upscalemodelloader.load_model(model_name="ESRGAN/4x_NMKD-Siax_200k.pth")
97
 
98
 
 
258
  image=get_value_at_index(upscaled_image, 0),
259
  )
260
 
261
+ saved_image = local_save.process_images(
262
+ prefix="advance_blur",
263
+ file_format="JPEG",
264
  images=get_value_at_index(final_image, 0),
265
  )
266