Add app.py

.gitignore

@@ -0,0 +1,2 @@
+env
+./env

app.py

@@ -0,0 +1,49 @@
+import gradio as gr
+from diffusers import StableDiffusionInpaintPipeline
+import torch
+from PIL import Image, ImageDraw
+# Load a pre-trained Stable Diffusion inpainting model
+pipe = StableDiffusionInpaintPipeline.from_pretrained("runwayml/stable-diffusion-inpainting", torch_dtype=torch.float16)
+pipe.to("cuda")
+
+# Function to transform mugshot into a professional portrait
+def transform_to_professional(image):
+    # Define prompt for a professional portrait
+    prompt = "a professional headshot of a person, studio lighting, corporate background, highly detailed, sharp focus"
+    negative_prompt = "blurry, low resolution, distorted, messy background"
+
+    # Resize and center the input image
+    image = image.resize((512, 512))
+    
+    # Add a mask (full image editing for simplicity)
+    mask = Image.new("L", image.size, 255)
+
+    # Generate professional portrait
+    result = pipe(
+        prompt=prompt,
+        image=image,
+        mask_image=mask,
+        num_inference_steps=50,
+        guidance_scale=7.5,
+        negative_prompt=negative_prompt,
+    )
+
+    # Return the generated image
+    return result.images[0]
+
+# Gradio Interface
+with gr.Blocks() as demo:
+    gr.Markdown("# Mugshot to Professional Headshot Converter")
+    gr.Markdown("Upload a mugshot, and we'll transform it into a professional headshot using Stable Diffusion.")
+
+    with gr.Row():
+        with gr.Column():
+            input_image = gr.Image(type="pil", label="Upload Mugshot")
+            submit_button = gr.Button("Transform")
+        with gr.Column():
+            output_image = gr.Image(type="pil", label="Professional Headshot")
+
+    submit_button.click(transform_to_professional, inputs=[input_image], outputs=[output_image])
+
+# Launch Gradio Interface
+demo.launch()

requirements.txt

@@ -0,0 +1,71 @@
+accelerate==1.1.1
+aiofiles==23.2.1
+annotated-types==0.7.0
+anyio==4.6.2.post1
+argcomplete==3.5.1
+certifi==2024.8.30
+charset-normalizer==3.4.0
+click==8.1.7
+colorama==0.4.6
+diffusers==0.31.0
+fastapi==0.115.5
+ffmpy==0.4.0
+filelock==3.16.1
+fsspec==2024.10.0
+gradio==5.6.0
+gradio_client==1.4.3
+h11==0.14.0
+httpcore==1.0.7
+httpx==0.27.2
+huggingface-hub==0.26.2
+idna==3.10
+importlib_metadata==8.5.0
+Jinja2==3.1.4
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+mdurl==0.1.2
+mpmath==1.3.0
+networkx==3.4.2
+numpy==2.1.3
+orjson==3.10.11
+packaging==24.1
+pandas==2.2.3
+pillow==11.0.0
+pipx==1.7.1
+platformdirs==4.3.6
+psutil==6.1.0
+pydantic==2.9.2
+pydantic_core==2.23.4
+pydub==0.25.1
+Pygments==2.18.0
+python-dateutil==2.9.0.post0
+python-multipart==0.0.12
+pytz==2024.2
+PyYAML==6.0.2
+regex==2024.11.6
+requests==2.32.3
+rich==13.9.4
+ruff==0.7.4
+safehttpx==0.1.1
+safetensors==0.4.5
+semantic-version==2.10.0
+setuptools==75.1.0
+shellingham==1.5.4
+six==1.16.0
+sniffio==1.3.1
+starlette==0.41.2
+sympy==1.13.1
+tokenizers==0.20.3
+tomlkit==0.12.0
+torch==2.5.1
+tqdm==4.67.0
+transformers==4.46.2
+typer==0.13.0
+typing_extensions==4.12.2
+tzdata==2024.2
+urllib3==2.2.3
+userpath==1.9.2
+uvicorn==0.32.0
+websockets==12.0
+wheel==0.44.0
+zipp==3.21.0