1

app.py

@@ -16,14 +16,13 @@ def image_to_data_uri(image_path):
     return f"data:{mime};base64,{data}"
 
 
-def edit_images(*args):
-    *imgs, prompt, token = args
+def edit_images(img1, img2, img3, img4, img5, prompt, token: gr.OAuthToken | None = None):
     if not token:
         raise gr.Error("Please sign in with your Hugging Face account first.")
     if not prompt or prompt.strip() == "":
         raise gr.Error("Please enter a prompt.")
 
-    images = [img for img in imgs if img is not None]
+    images = [img for img in [img1, img2, img3, img4, img5] if img is not None]
     if len(images) == 0:
         raise gr.Error("Please upload at least one image.")