Update app.py

app.py

@@ -11,6 +11,7 @@ from huggingface_hub import hf_hub_download
 from IP_Composer.IP_Adapter.ip_adapter import IPAdapterXL
 from IP_Composer.perform_swap import compute_dataset_embeds_svd, get_modified_images_embeds_composition
 import spaces
+import random
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -195,14 +196,16 @@ def process_and_display(
     concept_image2=None, concept_name2=None,
     concept_image3=None, concept_name3=None,
     rank1=30, rank2=30, rank3=30,
-    prompt=None, scale=1.0, seed=420
+    prompt=None, scale=1.0, seed=0
 ):
-    """Wrapper for process_images that handles UI updates"""
     if base_image is None:
-        return None, "Please upload a base image"
+        raise gr.Error("please upload a base image")
     
     if concept_image1 is None:
-        return None, "Please upload at least one concept image"
+        raise gr.Error("choose at least one concept image")
+
+    if concept_image1 is None:
+        raise gr.Error("choose at least one concept type")
     
     modified_images = process_images(
         base_image, 
@@ -213,10 +216,6 @@ def process_and_display(
         prompt, scale, seed
     )
     
-    # # Clean up memory
-    # torch.cuda.empty_cache()
-    # gc.collect()
-    
     return modified_images
 
 # UI CSS