initial commit

.gitignore

@@ -0,0 +1,34 @@
+# Ignore outputs (generated images)
+outputs/
+*.png
+
+# Ignore model weights (handled separately, e.g. via Git LFS or manual upload)
+model/*.pkl
+
+# Python cache
+__pycache__/
+*.py[cod]
+*.so
+
+# Jupyter notebooks checkpoint
+.ipynb_checkpoints/
+
+# System files
+.DS_Store
+Thumbs.db
+
+# Environment or build artifacts
+env/
+venv/
+*.egg-info/
+dist/
+build/
+.cache/
+*.log
+
+# VSCode and IDE settings
+.vscode/
+.idea/
+
+# Hugging Face specific
+*.sagemaker/

app.py

@@ -0,0 +1,44 @@
+import gradio as gr
+import subprocess
+import os
+import random
+from PIL import Image
+
+# Paths
+OUTPUT_DIR = "outputs"
+MODEL_PATH = "model/network-snapshot-000020.pkl"  # Adjusted for local Hugging Face repo structure
+
+# Ensure the output directory exists
+os.makedirs(OUTPUT_DIR, exist_ok=True)
+
+# Function to generate images using StyleGAN3
+def generate_images():
+    command = f"python stylegan3/gen_images.py --outdir={OUTPUT_DIR} --trunc=1 --seeds='3-5,7,9,12-14,16-26,29,31,32,34,40,41' --network={MODEL_PATH}"
+    try:
+        subprocess.run(command, shell=True, check=True)
+    except subprocess.CalledProcessError as e:
+        return f"Error generating images: {e}"
+
+# Function to select 5 random images
+def get_random_images():
+    image_files = [f for f in os.listdir(OUTPUT_DIR) if f.endswith(".png")]
+    if len(image_files) < 5:
+        generate_images()
+        image_files = [f for f in os.listdir(OUTPUT_DIR) if f.endswith(".png")]
+    random_images = random.sample(image_files, min(5, len(image_files)))
+    return [Image.open(os.path.join(OUTPUT_DIR, img)) for img in random_images]
+
+# Gradio function
+def generate_and_display():
+    generate_images()
+    return get_random_images()
+
+# UI
+with gr.Blocks() as demo:
+    gr.Markdown("# 🎨 AI-Generated Clothing Designs - Dresses")
+    generate_button = gr.Button("Generate New Designs")
+    output_gallery = gr.Gallery(label="Generated Designs", columns=5, rows=2)
+    generate_button.click(fn=generate_and_display, outputs=output_gallery)
+
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -0,0 +1,7 @@
+gradio==4.26.0
+Pillow==10.2.0
+torch==2.0.1
+torchvision==0.15.2
+torchaudio==2.0.2
+ninja==1.11.1
+