Update app.py

app.py

@@ -3,32 +3,29 @@ from PIL import Image
 import os
 # Import your core functions (estimate_pose, segment_clothing, inpaint_clothing, change_clothing) from your main script
 from main_code_script import change_clothing  # Replace your_main_script
-
-def predict(image_path, clothing_prompt):
+def predict(image_path, garment_image_path): # Changed input
     """
     The prediction function for Gradio.
     """
     try:
-        modified_image = change_clothing(image_path, clothing_prompt)
+        modified_image = change_clothing(image_path, garment_image_path) # Changed input
         if modified_image:
             return modified_image
         else:
-            return "Failed to change clothing. Please check the image and prompt."
+            return "Failed to change clothing.  Please check the images."
     except Exception as e:
         return f"Error: {e}"
-        
 # Create the Gradio interface
 iface = gr.Interface(
     fn=predict,
     inputs=[
-        gr.Image(type="filepath", label="Input Image"),
-        gr.Textbox(lines=2, label="Clothing Description (e.g., a blue dress)"),
+        gr.Image(type="filepath", label="Input Image (Person)"), # Changed label
+        gr.Image(type="filepath", label="Garment Image"),  # Added input
     ],
     outputs=gr.Image(type="pil", label="Modified Image"),
     title="AI Clothing Changer",
-    description="Try on different clothes with AI!",
+    description="Try on different clothes with AI by uploading a garment image!", # Changed description
 )
-
 # Launch the Gradio interface
 if __name__ == "__main__":
     iface.launch()