Update app.py

app.py

@@ -192,7 +192,7 @@ def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denois
                 )
 
                 prompt = "a photo of " + garment_des
-                negative_prompt = "monochrome, lowres, bad anatomy, worst quality, low quality , change color"
+                negative_prompt = "monochrome, lowres, bad anatomy, worst quality, low quality"
                 if not isinstance(prompt, list):
                     prompt = [prompt] * 1
                 if not isinstance(negative_prompt, list):
@@ -220,7 +220,7 @@ def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denois
                     negative_pooled_prompt_embeds=negative_pooled_prompt_embeds.to(device, torch.float16),
                     num_inference_steps=denoise_steps,
                     generator=generator,
-                    strength=1.5,
+                    strength=1.0,
                     pose_img=pose_img.to(device, torch.float16),
                     text_embeds_cloth=prompt_embeds_c.to(device, torch.float16),
                     cloth=garm_tensor.to(device, torch.float16),
@@ -229,7 +229,7 @@ def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denois
                     height=1024,
                     width=768,
                     ip_adapter_image=garm_img.resize((768, 1024)),
-                    guidance_scale=1.5,
+                    guidance_scale=4.0,
                 )[0]
 
     if is_checked_crop:
@@ -314,6 +314,31 @@ def tryon():
         'mask_image': mask_base64
     })
 
+
+# Route index
+@app.route('/', methods=['GET'])
+def index():
+
+    # Renvoyer l'image
+    try:
+        return 'Welcome to IDM VTON API'
+    except FileNotFoundError:
+        return jsonify({'error': 'Image not found'}), 404
+
+# Route pour récupérer l'image générée
+@app.route('/api/get_image/<image_id>', methods=['GET'])
+def get_image(image_id):
+    # Construire le chemin complet de l'image
+    image_path = image_id  # Assurez-vous que le nom de fichier correspond à celui que vous avez utilisé lors de la sauvegarde
+
+    # Renvoyer l'image
+    try:
+        return send_file(image_path, mimetype='image/webp')
+    except FileNotFoundError:
+        return jsonify({'error': 'Image not found'}), 404
+		
+		
+
 @spaces.GPU 
 def generate_mask(human_img, categorie='upper_body'):
     device = "cuda"
@@ -362,27 +387,7 @@ def generate_mask_api():
         logging.error(f"Error generating mask: {e}")
         return jsonify({'error': str(e)}), 500
 
-# Route index
-@app.route('/', methods=['GET'])
-def index():
 
-    # Renvoyer l'image
-    try:
-        return 'Welcome to IDM VTON API'
-    except FileNotFoundError:
-        return jsonify({'error': 'Image not found'}), 404
-
-# Route pour récupérer l'image générée
-@app.route('/api/get_image/<image_id>', methods=['GET'])
-def get_image(image_id):
-    # Construire le chemin complet de l'image
-    image_path = image_id  # Assurez-vous que le nom de fichier correspond à celui que vous avez utilisé lors de la sauvegarde
-
-    # Renvoyer l'image
-    try:
-        return send_file(image_path, mimetype='image/webp')
-    except FileNotFoundError:
-        return jsonify({'error': 'Image not found'}), 404
 
 if __name__ == "__main__":
     app.run(debug=False, host="0.0.0.0", port=7860)