Update app.py

app.py

@@ -146,10 +146,16 @@ def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denois
 
     with torch.no_grad():
         with torch.cuda.amp.autocast():
+            if not garment_des or not isinstance(garment_des, str):
+                garment_des = "a garment"
+    
             prompt = "model is wearing " + garment_des
             negative_prompt = "monochrome, lowres, bad anatomy, worst quality, low quality"
-            prompt_embeds, neg_embeds, pooled_prompt_embeds, neg_pooled_prompt_embeds = pipe.encode_prompt(
-                prompt, 1, True, negative_prompt)
+
+            if not garment_des or not isinstance(garment_des, str):
+                garment_des = "a garment"
+                
+            prompt_embeds, neg_embeds, pooled_prompt_embeds, neg_pooled_prompt_embeds = pipe.encode_prompt([prompt], 1, True, [negative_prompt])
 
             prompt_c = "a photo of " + garment_des
             prompt_embeds_c, _, _, _ = pipe.encode_prompt(