Update app.py

app.py

@@ -1,13 +1,10 @@
 import gradio as gr
-from huggingface_hub import hf_hub_download
+import spaces
 from diffusers import DiffusionPipeline
 
-# Load the pre-trained model
-model_id = "stabilityai/sdxl-turbo"
-model_path = hf_hub_download(repo_id=model_id, repo_type="model")
 pipe = DiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float32)
 
-# Define the Gradio interface
+@spaces.GPU(duration=120)
 def generate_image(prompt):
     # Run the diffusion model to generate an image
     image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5)
@@ -18,7 +15,8 @@ gr_interface = gr.Interface(
     inputs="text",
     outputs="image",
     title="Real-time Image Generation with Diffusion",
-    description="Enter a prompt to generate an image"
+    description="Enter a prompt to generate an image",
+    theme="soft"
 )
 
 # Launch the Gradio app