Add application file

app.py

@@ -4,15 +4,13 @@ import torch
 
 model_id = "runwayml/stable-diffusion-v1-5"
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
-# pipe = pipe.to("cuda")
+pipe = pipe.to("cuda")
 
-prompt = "a photo of an astronaut riding a horse on mars"
-    
+# prompt = "a photo of an astronaut riding a horse on mars"
 # image.save("astronaut_rides_horse.png")
 
-
-def infer(prompt):
-    image = pipe(prompt).images[0]  
+def infer(Prompt):
+    image = pipe(Prompt).images[0]  
     
     return image
 

requirements.txt

@@ -1,3 +1,4 @@
 git+https://github.com/huggingface/transformers.git
 git+https://github.com/huggingface/diffusers.git
-torch
+torch
+accelerate