Update space

app.py

@@ -2,7 +2,7 @@ import spaces
 import gradio as gr
 import numpy as np
 import torch
-import random
+from PIL import Image
 from diffusers import DDPMScheduler, StableDiffusionPipeline, DDIMScheduler, UNet2DConditionModel
 from diffusers import StableDiffusionInstructPix2PixPipeline, LCMScheduler
 
@@ -23,7 +23,7 @@ MAX_IMAGE_SIZE = 1024
 
 @spaces.GPU(duration=30)
 def infer(image, edit_instruction, guidance_scale, image_guidance_scale, n_steps):
-
+    image = Image.fromarray(image).resize((512, 512))
     image = pipe(prompt=edit_instruction, 
              image=image,
              num_inference_steps=n_steps, 
@@ -50,7 +50,7 @@ with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(
             f"""
-        # ⚡ Instruct-p2p with Consistency Distillation⚡ 
+        # ⚡ Instruct-pix2pix with Consistency Distillation⚡ 
         Currently running on {power_device}
         """
         )