Update app.py

app.py

@@ -1,3 +1,15 @@
 import gradio as gr
 
-gr.Interface.load("models/andite/pastel-mix").launch()
+gr.Interface.load("models/andite/pastel-mix").launch()
+
+from diffusers import StableDiffusionPipeline
+import torch
+
+model_id = "andite/pastel-mix"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+pipe = pipe.to("cuda")
+
+prompt = "hatsune_miku"
+image = pipe(prompt).images[0]
+
+image.save("./hatsune_miku.png")