Update app.py

app.py

@@ -326,6 +326,9 @@ def infer_relight(
     if len(pil_images) == 0:
         raise gr.Error("Please upload an image first.")
 
+    if height != 0 and width != 0:
+        pil_images[0] = pil_images[0].resize((width, height), Image.LANCZOS)
+
     result = pipe(
         image=pil_images,
         prompt=final_prompt,