Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import torch
 from diffusers import StableDiffusionXLPipeline, AutoencoderKL, KDPM2AncestralDiscreteScheduler
 from huggingface_hub import hf_hub_download
+import spaces
 from PIL import Image
 import requests
 from translatepy import Translator
@@ -28,8 +29,6 @@ JS = """function () {
   }
 }"""
 
-device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-
 # Load VAE component
 vae = AutoencoderKL.from_pretrained(
     vae_model, 
@@ -37,12 +36,14 @@ vae = AutoencoderKL.from_pretrained(
 )
 
 # Ensure model and scheduler are initialized in GPU-enabled function
-pipe = StableDiffusionXLPipeline.from_pretrained(model, vae=vae).to(device)
+if torch.cuda.is_available():
+    pipe = StableDiffusionXLPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16).to("cuda")
 
 pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 
 
 # Function 
+@spaces.GPU()
 def generate_image(
     prompt,
     negative="low quality",
@@ -91,7 +92,7 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
     img = gr.Image(label='Mobius Generated Image')
     with gr.Accordion("Advanced Options", open=False):
         with gr.Row():
-            negative = gr.Textbox(label="Negative prompt", value="low quality, ugly, blurry")
+            negative = gr.Textbox(label="Negative prompt", value="low quality")
         with gr.Row():
             width = gr.Slider(
                 label="Width",