more readme updates; demo cleanup

README.md

@@ -21,7 +21,7 @@ A tool for improving facial consistency and quality in AI-generated images. Dram
 ### Prerequisites
 - Python 3.11 or higher
 - 1 GPU with 48GB VRAM
-- At least 50GB of free disk space
+- At least 60GB of free disk space
 
 ### Setup
 
@@ -33,7 +33,7 @@ A tool for improving facial consistency and quality in AI-generated images. Dram
      export HUGGINGFACE_TOKEN=your_token_here
      export HF_HOME=/path/to/your/huggingface_cache
      ```
-   - Models will be downloaded to `$HF_HOME` and symlinked to `./ComfyUI/models/`
+   - Models will be downloaded to `$HF_HOME` and symlinked to `FaceEnhance/ComfyUI/models/`
 
 2. Create the virtual environment:
    ```
@@ -73,9 +73,6 @@ A tool for improving facial consistency and quality in AI-generated images. Dram
    - `--input` (str): Path to the input image.
    - `--ref` (str): Path to the reference face image.
    - `--output` (str): Path to save the output image.
-   - `--crop` (store_true): Flag to face crop the reference image. Default: False.
-   - `--upscale` (store_true): Flag to upscale the (cropped) reference image. Default: False.
-   - `--caption` (store_true): Flag to caption the input image. Default: False.
    - `--id_weight` (float): Face ID weight. Default: 0.75.
    </details>
 
@@ -89,15 +86,15 @@ A simple web interface for the face enhancement workflow.
 
 ## Running on ComfyUI
 
-Using the ComfyUI workflows is the fastest way to get started. Run `python run_comfy.py`. There are two workflows:
-- `./workflows/FaceEnhancementProd.json` for face enhancement
-- `./workflows/FaceEmbedDist.json` for computing the [face embedding distance](https://github.com/cubiq/ComfyUI_FaceAnalysis)
+Run `python run_comfy.py`. There are two workflows:
+- `FaceEnhance/workflows/FaceEnhancementProd.json` for face enhancement
+- `FaceEnhance/workflows/FaceEmbedDist.json` for computing the [face embedding distance](https://github.com/cubiq/ComfyUI_FaceAnalysis)
 
 
 ### Notes
 - The script and demo run a ComfyUI server ephemerally
 - Gradio demo is faster than the script because the models remain loaded in memory and ComfyUI server is booted up.
-- All images are saved in `./ComfyUI/input/scratch/`
+- Images are saved in `FaceEnhance/ComfyUI/input/scratch/`
 - `FaceEnhancementProd.py` was created with the [ComfyUI-to-Python-Extension](https://github.com/pydn/ComfyUI-to-Python-Extension) and re-engineered for efficiency and function.
 - Face cropping, upscaling, and captioning are unavailable; these will be added in an update.
 

demo.py

@@ -98,17 +98,15 @@ def enhance_face_gradio(input_image, ref_image):
     return result_img
 
 def create_gradio_interface():
-    with gr.Blocks(title="Face Enhancement Demo") as demo:
+    with gr.Blocks(title="Face Enhancement") as demo:
         gr.Markdown("""
-        # Face Enhancement
+        # Face Enhance
         ### Instructions
         1. Upload the target image you want to enhance
-        2. Upload a high-quality reference face image
+        2. Upload a high-quality face image
         3. Click 'Enhance Face'
 
-        Processing takes about 60 seconds. Due to the constraints of this demo, face cropping and upscaling are not applied to the reference image.
-
-        For more information, check out my [blog post](https://rishidesai.github.io/posts/face-enhancement-techniques/).
+        Processing takes about 60 seconds.
         """, elem_id="instructions")
 
         gr.Markdown("---")
@@ -128,8 +126,10 @@ def create_gradio_interface():
             outputs=output_image,
             queue=True  # Enable queue for sequential processing
         )
-
-        gr.Markdown("## Examples\nClick on an example to load the images into the interface.")
+        gr.Markdown("""
+        ## Examples
+        Click on an example to load the images into the interface.
+        """)
         example_inps = [
             ["examples/dany_gpt_1.png", "examples/dany_face.jpg"],
             ["examples/dany_gpt_2.png", "examples/dany_face.jpg"],
@@ -139,6 +139,13 @@ def create_gradio_interface():
         ]
         gr.Examples(examples=example_inps, inputs=[input_image, ref_image], outputs=output_image)
 
+        gr.Markdown("""
+        ## Notes
+        Check out the code [here](https://github.com/RishiDesai/FaceEnhance) and see my [blog post](https://rishidesai.github.io/posts/face-enhancement-techniques/) for more information.
+        
+        Due to the constraints of this demo, face cropping and upscaling are not applied to the reference image.
+        """)
+
     # Launch the Gradio app with queue
     demo.queue(max_size=99)
     

test.py

@@ -7,8 +7,6 @@ def parse_args():
     parser = argparse.ArgumentParser(description='Face Enhancement Tool')
     parser.add_argument('--input', type=str, required=True, help='Path to the input image')
     parser.add_argument('--ref', type=str, required=True, help='Path to the reference image')
-    parser.add_argument('--crop', action='store_true', help='Whether to crop the image')
-    parser.add_argument('--upscale', action='store_true', help='Whether to upscale the image')
     parser.add_argument('--output', type=str, required=True, help='Path to save the output image')
     parser.add_argument('--id_weight', type=float, default=0.75, help='face ID weight')
     args = parser.parse_args()
@@ -41,7 +39,7 @@ def create_scratch_dir():
 
     return new_dir
 
-def process_face(input_path, ref_path, crop=False, upscale=False, output_path=None, id_weight=0.75):
+def process_face(input_path, ref_path, output_path=None, id_weight=0.75):
     """
     Process a face image using the given parameters.
 
@@ -83,8 +81,6 @@ def main():
     return process_face(
         input_path=args.input,
         ref_path=args.ref,
-        crop=args.crop,
-        upscale=args.upscale,
         output_path=args.output,
         id_weight=args.id_weight
     )