model2 commited on
Commit
4bb9b6b
·
1 Parent(s): 2941d7e

image sizing

Browse files
Files changed (1) hide show
  1. app.py +1 -8
app.py CHANGED
@@ -1,10 +1,3 @@
1
- # TODO: Replace face images with face model to stop creating it every time
2
- # TODO: Figure out .cache error
3
- # TODO: Figure out comfyui-frontend error
4
- # TODO: Preload retinaface_resnet50
5
- # TODO: Preload nsfw model
6
- # TODO: Max out final image size as original image size
7
- # TODO: Double-check inclusion of all necessary custom nodes in repo
8
  # TODO: UI/UX: Better display on mobile so folks don't miss the final output
9
  # TODO: Upgrade gradio
10
 
@@ -90,7 +83,7 @@ def import_custom_nodes() -> None:
90
  # Preload nodes, models.
91
  import_custom_nodes()
92
  loadimage = NODE_CLASS_MAPPINGS["LoadImage"]()
93
- getimagesize = NODE_CLASS_MAPPINGS["GetImageSize"]()
94
  upscalemodelloader = NODE_CLASS_MAPPINGS["UpscaleModelLoader"]()
95
  reactorloadfacemodel = NODE_CLASS_MAPPINGS["ReActorLoadFaceModel"]()
96
  FACE_MODEL = reactorloadfacemodel.load_model(
 
 
 
 
 
 
 
 
1
  # TODO: UI/UX: Better display on mobile so folks don't miss the final output
2
  # TODO: Upgrade gradio
3
 
 
83
  # Preload nodes, models.
84
  import_custom_nodes()
85
  loadimage = NODE_CLASS_MAPPINGS["LoadImage"]()
86
+ getimagesize = NODE_CLASS_MAPPINGS["GetImageSize+"]()
87
  upscalemodelloader = NODE_CLASS_MAPPINGS["UpscaleModelLoader"]()
88
  reactorloadfacemodel = NODE_CLASS_MAPPINGS["ReActorLoadFaceModel"]()
89
  FACE_MODEL = reactorloadfacemodel.load_model(