test

app.py

@@ -1,3 +1,4 @@
+import spaces
 from diffusers import StableDiffusionXLPipeline
 from pydantic import BaseModel
 from PIL import Image
@@ -6,7 +7,7 @@ import torch
 import uuid
 import io
 import os
-import spaces  # Assuming spaces is a valid module with GPU decorator
+
 
 # Load your model
 pipe = StableDiffusionXLPipeline.from_pretrained(
@@ -28,7 +29,7 @@ def generate_and_save_image(prompt, negative_prompt=''):
 
     # Save generated image locally
     os.makedirs('generated_images', exist_ok=True)
-    image.save(image_path, format='PNG')
+    image.save(image_path, format='jpeg/jpg')
 
     # Return the path of the saved image to display in Gradio interface
     return image_path

requirements.txt

@@ -5,4 +5,5 @@ torch
 uuid
 pydantic
 Pillow
-accelerate
+accelerate
+spaces