changed name

.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
+*.png filter=lfs diff=lfs merge=lfs -text
+*.jpg filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,5 +1,5 @@
 ---
-title: Blueberry V0
+title: AYNA 1.0
 emoji: 🚀
 colorFrom: purple
 colorTo: yellow

app.py

@@ -10,7 +10,6 @@ import torch
 
 # Add the parent directory to the sys.path
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
-from ckpt.pipeline import BlueberryPipeline
 
 # Download entire repository
 def download_repo(repo_id=os.getenv("MODEL_REPO_ID"), local_dir="./ckpt"):
@@ -41,6 +40,8 @@ def download_repo(repo_id=os.getenv("MODEL_REPO_ID"), local_dir="./ckpt"):
 # Download model repository
 repo_path = download_repo()
 
+from ckpt.pipeline import BlueberryPipeline
+
 # Simplified Model Cache class for loading and storing only the pipeline
 class ModelCache:
     def __init__(self):
@@ -59,7 +60,7 @@ class ModelCache:
 # Initialize model cache
 model_cache = ModelCache()
 
-@spaces.GPU
+@spaces.GPU(duration=60)
 def virtual_tryon(garment_img, person_img, garment_type, sleeve_length, garment_length):
     """
     Perform virtual try-on with the given garment on the target person
@@ -140,21 +141,47 @@ with gr.Blocks(css=css) as demo:
         
         with gr.Row():
             with gr.Column(scale=2.5):
+                with gr.Column():
+                    with gr.Row():
+                        # add example images for garment and humans
+                        example_garments = [
+                            "examples/garments/garment-1.png",
+                            "examples/garments/garment-2.jpg",
+                            "examples/garments/garment-3.jpg"
+                        ]
+                        example_humans = [
+                            "examples/humans/human-1.jpg",
+                            "examples/humans/human-2.jpg",
+                            "examples/humans/human-3.jpg"
+                        ]
+                        gr.Examples(
+                            examples=example_garments,
+                            inputs=garment_input,
+                            label="Garment Examples",
+                            examples_per_page=3
+                        )
+                        gr.Examples(
+                            examples=example_humans,
+                            inputs=person_input,
+                            label="Person Examples",
+                            examples_per_page=3
+                        )
                 # Add radio buttons for different garment types
-                garment_type = gr.Radio(choices=["upper", "lower", "full-body"], label="Garment Type", value="upper")
-                sleeve_length = gr.Radio(
-                    choices=["3/4 sleeve", "cap sleeve", "short sleeve", "long sleeve", "sleeveless", "ignore"], 
-                    label="Sleeve Length", 
-                    visible=True, 
-                    interactive=True,
-                    info="Choose 'ignore' if you are not sure"
-                )
-                garment_length = gr.Radio(
-                    choices=["crop length", "hip length", "waist length", "tunic length", "thigh length", "knee length", "ignore"],
-                    label="Garment Length",
-                    interactive=True,
-                    info="Choose 'ignore' if you are not sure"
-                )
+                with gr.Column():
+                    garment_type = gr.Radio(choices=["upper", "lower", "full-body"], label="Garment Type", value="upper")
+                    sleeve_length = gr.Radio(
+                        choices=["3/4 sleeve", "cap sleeve", "short sleeve", "long sleeve", "sleeveless", "ignore"], 
+                        label="Sleeve Length", 
+                        visible=True, 
+                        interactive=True,
+                        info="Choose 'ignore' if you are not sure"
+                    )
+                    garment_length = gr.Radio(
+                        choices=["crop length", "hip length", "waist length", "tunic length", "thigh length", "knee length", "ignore"],
+                        label="Garment Length",
+                        interactive=True,
+                        info="Choose 'ignore' if you are not sure"
+                    )
                 
                 # Update sleeve_length visibility based on garment_type
                 def update_sleeve_visibility(garment):