Update app.py

app.py

@@ -5,18 +5,19 @@ import gradio as gr
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 pipe = StableDiffusionXLPipeline.from_pretrained(
-    "stabilityai/sdxl-turbo",
+    "stabilityai/stable-diffusion-xl-base-1.0",
     torch_dtype=torch.float16 if device == "cuda" else torch.float32,
+    use_safetensors=True,
     variant="fp16" if device == "cuda" else None
 ).to(device)
 
 def generate_image(prompt):
-    image = pipe(prompt=prompt, guidance_scale=0.0, num_inference_steps=1).images[0]
+    image = pipe(prompt=prompt, num_inference_steps=30, guidance_scale=7.5).images[0]
     return image
 
 gr.Interface(
     fn=generate_image,
     inputs="text",
     outputs="image",
-    title="Sanjay's Image Generator AI"
+    title="Sanjay's Perfect Image Generator"
 ).launch()