Update app.py

app.py

@@ -15,12 +15,9 @@ css = """
     }
 """
 
-import os
-os.makedirs("cache", exist_ok=True)
-
 DEVICE = 'cuda'
 model_id = "Onodofthenorth/SD_PixelArt_SpriteSheet_Generator"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, cache_dir="cache", torch_dtype=torch.float16)
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
 pipe.to("cuda")
 
 
@@ -39,7 +36,7 @@ with gr.Blocks(css=css) as API:
         
     with gr.Column():
         inputs=gr.TextArea(label="Prompt", placeholder="Prompt")
-        outputs=gr.Image(label="Ouput Image", type='pil', height=500)
+        outputs=gr.Image(label="Ouput Image", type='pil', elem_id="img-display-output")
     generate_btn = gr.Button(value="Generate")
     generate_btn.click(generate_sprite, inputs=inputs, outputs=outputs, api_name="generate_mesh")