Create app.py

app.py

@@ -0,0 +1,8 @@
+from diffusers import AutoPipelineForText2Image
+from diffusers import DiffusionPipeline
+
+pipe = DiffusionPipeline.from_pretrained("shuttleai/shuttle-3.1-aesthetic",
+                                         torch_dtype=torch.float16,USE_SAFETENSORS=True).to("cuda")
+
+prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
+image = pipe(prompt).images[0]