hi

app.py

@@ -1,11 +1,13 @@
 import spaces
 import gradio as gr
 import torch
-from diffusers import DiffusionPipeline
+from diffusers import DiffusionPipeline, AutoencoderKL
 import rembg
 
 model_id = "dataautogpt3/OpenDalleV1.1"
+vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 pipe = DiffusionPipeline.from_pretrained(model_id,
+                                         vae=vae,
                                          torch_dtype=torch.float16,
                                         use_safetensors=True,
                                         variant="fp16")
@@ -24,7 +26,7 @@ _TITLE = "Shoe Generator"
 with gr.Blocks(_TITLE) as ShoeGen:
     with gr.Row():
         with gr.Column():
-            prompt = gr.Textbox(label="Enter a discription of a shoe", value="no background, side view, minimalist shot")
+            prompt = gr.Textbox(label="Enter a discription of a shoe")
             neg_prompt = gr.Textbox(label="Enter a negative prompt", value="low quality, watermark, ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, extra limbs, body out of frame, blurry, bad anatomy, blurred, watermark, grainy, signature, cut off, draft, closed eyes, text, logo")
             button_gen = gr.Button("Generate Image")
         with gr.Column():