Update app.py

app.py

@@ -193,6 +193,8 @@ def infer_with_lora(input_image, prompt, selected_index, custom_lora, seed=42, r
             image=input_image, 
             prompt=prompt,
             guidance_scale=guidance_scale,
+            width=input_image.size[0],
+            height=input_image.size[1],
             generator=torch.Generator().manual_seed(seed),
         ).images[0]