Update app.py

app.py

@@ -22,4 +22,25 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()
+
+    from diffusers import StableDiffusionPipeline
+import torch
+
+# Load model from Hugging Face Hub
+pipe = StableDiffusionPipeline.from_pretrained(
+    "runwayml/stable-diffusion-v1-5",  # You can use another model if needed
+    torch_dtype=torch.float16,
+    use_safetensors=True,
+    revision="fp16"
+)
+
+# Move to GPU if available
+pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
+
+# Generate an image
+prompt = "A Minecraft player standing on a hill during sunset, pixel art style"
+image = pipe(prompt).images[0]
+
+# Save the image
+image.save("minecraft_thumbnail.png")