Upload 11 files

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+display.png filter=lfs diff=lfs merge=lfs -text
+images/output_images/output2.png filter=lfs diff=lfs merge=lfs -text

FaceSwap.py

@@ -0,0 +1,146 @@
+#This is an example that uses the websockets api to know when a prompt execution is done
+#Once the prompt execution is done it downloads the images using the /history endpoint
+
+import websocket #NOTE: websocket-client (https://github.com/websocket-client/websocket-client)
+import uuid
+import json
+import urllib.request
+import urllib.parse
+
+server_address = "127.0.0.1:8188"
+client_id = str(uuid.uuid4())
+
+def queue_prompt(prompt):
+    p = {"prompt": prompt, "client_id": client_id}
+    data = json.dumps(p).encode('utf-8')
+    req =  urllib.request.Request("http://{}/prompt".format(server_address), data=data)
+    return json.loads(urllib.request.urlopen(req).read())
+
+def get_image(filename, subfolder, folder_type):
+    data = {"filename": filename, "subfolder": subfolder, "type": folder_type}
+    url_values = urllib.parse.urlencode(data)
+    with urllib.request.urlopen("http://{}/view?{}".format(server_address, url_values)) as response:
+        return response.read()
+
+def get_history(prompt_id):
+    with urllib.request.urlopen("http://{}/history/{}".format(server_address, prompt_id)) as response:
+        return json.loads(response.read())
+
+def get_images(ws, prompt):
+    prompt_id = queue_prompt(prompt)['prompt_id']
+    output_images = {}
+    while True:
+        out = ws.recv()
+        if isinstance(out, str):
+            message = json.loads(out)
+            if message['type'] == 'executing':
+                data = message['data']
+                if data['node'] is None and data['prompt_id'] == prompt_id:
+                    break #Execution is done
+        else:
+            continue #previews are binary data
+
+    history = get_history(prompt_id)[prompt_id]
+    for node_id in history['outputs']:
+        node_output = history['outputs'][node_id]
+        images_output = []
+        if 'images' in node_output:
+            for image in node_output['images']:
+                image_data = get_image(image['filename'], image['subfolder'], image['type'])
+                images_output.append(image_data)
+        output_images[node_id] = images_output
+
+    return output_images
+
+prompt_text = """
+{
+  "1": {
+    "inputs": {
+      "image": "/home/ml/Desktop/comfy_to_python/output.jpg",
+      "upload": "image"
+    },
+    "class_type": "LoadImage",
+    "_meta": {
+      "title": "Load Image"
+    }
+  },
+  "2": {
+    "inputs": {
+      "image": "/home/ml/Desktop/comfy_to_python/me.jpg",
+      "upload": "image"
+    },
+    "class_type": "LoadImage",
+    "_meta": {
+      "title": "Load Image"
+    }
+  },
+  "4": {
+    "inputs": {
+      "images": [      
+        "5",
+        0
+      ]
+    },
+    "class_type": "PreviewImage",
+    "_meta": {
+      "title": "Preview Image"
+    }
+  },
+  "5": {
+    "inputs": {
+      "enabled": true,
+      "swap_model": "inswapper_128.onnx",
+      "facedetection": "YOLOv5l",
+      "face_restore_model": "none",
+      "face_restore_visibility": 1,
+      "codeformer_weight": 1,
+      "detect_gender_input": "no",
+      "detect_gender_source": "no",
+      "input_faces_index": "0",
+      "source_faces_index": "0",
+      "console_log_level": 1,
+      "input_image": [
+        "1",
+        0
+      ],
+      "source_image": [
+        "2",
+        0
+      ]
+    },
+    "class_type": "ReActorFaceSwap",
+    "_meta": {
+      "title": "ReActor 🌌 Fast Face Swap"
+    }
+  }
+}
+"""
+
+prompt = json.loads(prompt_text)
+# #set the text prompt for our positive CLIPTextEncode
+# prompt["6"]["inputs"]["text"] = "your instruction here"
+
+prompt["1"]["inputs"]["image"] = "/home/ml/Desktop/comfy_to_python/66.jpg"
+prompt["2"]["inputs"]["image"] = "/home/ml/Desktop/comfy_to_python/me.jpg"
+
+# # If you have a group input face image change the number here (1,2,3,..) if single then put 0.
+# prompt["5"]["inputs"]["input_faces_index"] = ""
+
+# # If you have a group source face image change the number here (1,2,3,..) if single then put 0.
+# prompt["5"]["inputs"]["source_faces_index"] = ""
+
+ws = websocket.WebSocket()
+ws.connect("ws://{}/ws?clientId={}".format(server_address, client_id))
+images = get_images(ws, prompt)
+
+# Commented out code to display the output images:
+
+for node_id in images:
+    for image_data in images[node_id]:
+        from PIL import Image
+        import io
+        image = Image.open(io.BytesIO(image_data))
+        image.save("output1.jpg")
+        # image.show()
+
+

README.md

@@ -1,3 +1,140 @@
----
-license: apache-2.0
----
+---
+language:
+- en
+tags:
+- FaceSwap
+- Headshot
+- SmoothChange
+- FaceChange
+- SmoothSwap
+- SwapFace
+- FaceEditing
+
+license: apache-2.0
+---
+# FaceSwap - AI-Powered Headshot Generator
+
+**FaceSwap** is a flow designed to seamlessly swap faces between two images using deep learning models. This project is built on top of ComfyUI, a modular and flexible interface that enables easy integration and customization of AI workflows. By leveraging advanced facial recognition and manipulation technologies, FaceSwap allows users to replace one face with another in images while maintaining realistic features and alignment.
+
+FaceSwap utilizes models to detect key facial landmarks, map them to the target face, and generate a realistic face swap by considering aspects like lighting, expression, and pose. This flow is designed to provide an easy-to-use interface through ComfyUI, allowing users to swap faces effortlessly with minimal setup.
+
+<div align="center">
+  <img width="600" height="500" alt="foduucom/FaceSwap" src="https://huggingface.co/foduucom/Headshot_Generator-FaceSwap/resolve/main/display.png">
+</div>
+
+## Requirements
+
+To run **FaceSwap**, you'll need to install **ComfyUI** and set up a few dependencies. Follow the steps below for installation and setup.
+
+### Prerequisites
+- **Python 3.9 or higher** (Ensure Python 3.9+ is installed on your machine)
+- **NVIDIA GPU** (Recommended for faster processing, though CPU can also be used)
+
+---
+
+## Step-by-Step Installation
+
+1. Clone the ComfyUI Repository
+Start by cloning the main ComfyUI repository:
+
+```bash
+git clone https://github.com/comfyanonymous/ComfyUI.git
+cd ComfyUI
+```
+
+2. Install Dependencies
+Install all the required dependencies for ComfyUI:
+
+```bash
+pip install -r requirements.txt
+```
+
+3. Install **ComfyUI-Manager**
+To manage custom nodes, you'll need to install ComfyUI-Manager. Run the following commands inside the **ComfyUI/custom_nodes** directory:
+
+```bash
+cd ComfyUI/custom_nodes
+git clone https://github.com/ltdrdata/ComfyUI-Manager.git
+```
+
+After cloning the ComfyUI-Manager, **restart ComfyUI**.
+
+4. Install **Reactor Node for ComfyUI** by clicking on Manager -> Custom Nodes Manage
+<div align="center">
+  <img width="600" height="500" alt="foduucom/FaceSwap" src="https://huggingface.co/foduucom/Headshot_Generator-FaceSwap/resolve/main/node.png">
+</div>
+
+or you can clone the repository in the directory **ComfyUI/custom_nodes**
+```bash
+git clone https://github.com/Gourieff/comfyui-reactor-node.git
+```
+
+5. Download Pre-trained Models
+Download **inswapper_128.onnx** and place the downloaded model in the ComfyUI/models/insightface/ directory
+
+Note: If directory don't exist, create them manually.
+
+4. Start ComfyUI
+Run ComfyUI by executing the following command:
+
+```bash
+python3 FaceSwap.py
+```
+This will start ComfyUI, and you can access the interface by navigating to http://127.0.0.1:8188 in your browser.
+
+## Setting Up and Using the FaceSwap Workflow
+
+1. Clone the Headshot_Generator-FaceSwap Repository
+Clone this repository that contains the FaceSwap workflow and assets:
+
+```bash
+git clone https://huggingface.co/foduucom/Headshot_Generator-FaceSwap
+cd Headshot_Generator-FaceSwap
+```
+
+2. How to use
+Once ComfyUI is running, open the browser window (default: http://127.0.0.1:8188), and follow these steps:
+
+- Click on the **Load button** in the menu bar and select the **workflow.json** file from this repo you just cloned.
+- Upload the **Workflow.json** to start the face-swapping process.
+- Provide Input Images
+  - **Source Image:** The image containing the face you want to swap.
+  - **Target Image:** The image where the face will be swapped into.
+- Click **Queue Prompt** to initiate the face-swapping process. The AI model will process the images and generate the output.
+
+or you can use by python script provided in this repository:
+```bash 
+python3 FaceSwap.py
+
+#Remember change the input paths in script here :
+#prompt["1"]["inputs"]["image"] = "//put your input image"
+#prompt["2"]["inputs"]["image"] = "//put your source face image"
+
+```
+
+## Troubleshooting
+- ComfyUI Not Running: Ensure that all dependencies are installed correctly and that you’re using Python 3.9 or higher. If issues persist, check the ComfyUI GitHub repository for troubleshooting guides.
+
+- Missing Models: Ensure you’ve downloaded the required models (sam_vit_h_4b8939.pth and groundingdino_swint_ogc.pth) and placed them in the correct directories (ComfyUI/models/sams/ and ComfyUI/models/grounding-dino/).
+
+## Slow Performance: 
+Using a GPU is highly recommended for better performance. If you’re using a CPU, the processing time may be longer.
+
+## Compute Infrastructure
+
+## Hardware
+
+NVIDIA GeForce RTX 3060 card
+
+## Model Card Contact
+
+For inquiries and contributions, please contact us at [email protected].
+
+```bibtex
+@ModelCard{
+  author    = {Nehul Agrawal and
+               Priyal Mehta},
+  title     = {FaceSwap - AI-Powered Face Swap / Headshot Generator},
+  year      = {2024}
+}
+```

inswapper_128.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e4a3f08c753cb72d04e10aa0f7dbe3deebbf39567d4ead6dce08e98aa49e16af
+size 554253681