Update app.py

app.py

@@ -44,7 +44,7 @@ def generate_image(prompt, height, width, num_inference_steps, seed=42, randomiz
         width=int(width),
         num_inference_steps=int(num_inference_steps),
         guidance_scale=0.0,  # Guidance should be 0 for Turbo models
-        generator = torch.Generator(device=device).manual_seed(seed)
+        generator = torch.Generator(device="cuda").manual_seed(seed)
     ).images[0]
 
     image_no_lora = pipe_no_lora(
@@ -53,7 +53,7 @@ def generate_image(prompt, height, width, num_inference_steps, seed=42, randomiz
         width=int(width),
         num_inference_steps=int(num_inference_steps),
         guidance_scale=0.0,  # Guidance should be 0 for Turbo models
-        generator = torch.Generator(device=device).manual_seed(seed)
+        generator = torch.Generator(device="cuda").manual_seed(seed)
     ).images[0]
     
     return (image_no_lora,image), seed
@@ -67,9 +67,9 @@ examples = [
     ["Astronaut riding a horse on Mars, cinematic lighting, sci-fi concept art, highly detailed"],
     ["Portrait of a wise old wizard with a long white beard, holding a glowing crystal staff, magical forest background"],
 ]
-css = '''#col-container { max-width: 1000px; margin: 0 auto; }
+css = '''#col-container { max-width: 900px; margin: 0 auto; }
 .dark .progress-text{color: white !important}
-#examples{max-width: 1000px; margin: 0 auto; }'''
+#examples{max-width: 900px; margin: 0 auto; }'''
 # Build the Gradio interface
 with gr.Blocks() as demo: