updated for cpu

README.md

@@ -7,6 +7,11 @@ sdk: gradio
 sdk_version: "3.50.2"
 app_file: app.py
 pinned: false
+hardware: true
+resources:
+  cpu: 1
+  memory: "16Gi"
+  gpu: 1
 ---
 
 # Style-Guided Image Generation with Purple Enhancement

app.py

@@ -40,10 +40,14 @@ styles = {
 
 def load_pipeline():
     """Load and prepare the pipeline with all style embeddings"""
+    # Check if CUDA is available
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    dtype = torch.float16 if device == "cuda" else torch.float32
+
     pipe = StableDiffusionPipeline.from_pretrained(
         "CompVis/stable-diffusion-v1-4",
-        torch_dtype=torch.float16
-    ).to("cuda")
+        torch_dtype=dtype
+    ).to(device)
     
     # Load all embeddings
     for style_info in styles.values():
@@ -71,8 +75,9 @@ def generate_image(prompt, style, seed, apply_guidance, guidance_strength=0.5):
     # Get style info
     style_info = styles[style]
     
-    # Prepare generator
-    generator = torch.Generator("cuda").manual_seed(int(seed))
+    # Prepare generator with appropriate device
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    generator = torch.Generator(device).manual_seed(int(seed))
     
     # Create styled prompt
     styled_prompt = f"{prompt} {style_info['token']}"