add accelerate

app.py

@@ -7,7 +7,7 @@ from io import BytesIO
 import streamlit as st
 from diffusers import StableDiffusionPipeline
 import torch
- 
+from accelerate import Accelerator
 #model_id = "CompVis/stable-diffusion-v1-4"
 #pipe = StableDiffusionPipeline.from_pretrained(model_id)
  
@@ -15,6 +15,7 @@ import torch
 #pipe = pipe.to("cpu")
  
 image_html = ""
+accelerator = Accelerator()
  
 # Function to display an example image
 def display_example_image(url):
@@ -26,7 +27,8 @@ def display_example_image(url):
 def generate_images_using_huggingface_diffusers(text):
 #    pipe = StableDiffusionPipeline.from_pretrained("sd-dreambooth-library/cat-toy", torch_dtype=torch.float16)
     pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float32)    
-    pipe = pipe.to("cpu")
+    # pipe = pipe.to("cpu")
+    pipe = pipe.to(accelerator.device)
     prompt = text
     image = pipe(prompt).images[0]
     return image

requirements.txt

@@ -1,3 +1,4 @@
 diffusers==0.29.2
 transformers==4.42.3
-torch==2.3.1
+torch==2.3.1
+accelerate==0.31.0