fix cpu problems

app.py

@@ -10,12 +10,16 @@ device = "cpu"
 # model_id_or_path = "runwayml/stable-diffusion-v1-5"
 # takes too much time downloading, change to local path
 model_path = "."
-img2img_pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_path)
+img2img_pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_path, torch_dtype=torch.float32)
 img2img_pipe = img2img_pipe.to(device)
 
 
-def img2img_diff(prompt, img):
+def img2img_diff(prompt, pil_img):
+    img = pil_img.resize((768, 512))
     return img2img_pipe(prompt=prompt, image=img, strength=0.75, guidance_scale=7.5).images[0]
 
 
-gr.Interface(fn=img2img_diff, inputs=['text', 'image'], outputs='image')
+app = gr.Interface(fn=img2img_diff, inputs=[gr.Text(label="prompt text"), gr.Image(type='pil', label='draft image')],
+                   outputs=gr.Image(type='pil'))
+if __name__ == "__main__":
+    app.launch(debug=True)