Update app.py

app.py

@@ -53,7 +53,7 @@ examples = [
 model_id = "runwayml/stable-diffusion-v1-5"
 
 def main():
-    pipe = StableDiffusionSAGPipeline.from_pretrained(model_id)#, torch_dtype=torch.float16).to('cuda')
+    pipe = StableDiffusionSAGPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to('cuda')
 
     def generate(
         prompt: str,
@@ -68,7 +68,7 @@ def main():
         generator = torch.manual_seed(seed)
         ori_image = pipe(prompt, generator=generator, guidance_scale=cfg_scale, sag_scale=0.0).images[0]
         generator = torch.manual_seed(seed)
-        sag_image = pipe(prompt, generator=generator, guidance_scale=cfg_scale, sag_scale=0.75).images[0]
+        sag_image = pipe(prompt, generator=generator, guidance_scale=cfg_scale, sag_scale=sag_scale).images[0]
         return [ori_image, sag_image]
 
     def reset():