markdown and scheduler changes

app.py

@@ -39,8 +39,9 @@ class AttributionModel:
         is_cuda = False
         if torch.cuda.is_available():
             is_cuda = True
-
-        self.pipe = StableDiffusionPipeline.from_pretrained('stabilityai/stable-diffusion-2')#, safety_checker=None, torch_dtype=torch.float16)
+        
+        scheduler = EulerDiscreteScheduler.from_pretrained('stabilityai/stable-diffusion-2', subfolder="scheduler")
+        self.pipe = StableDiffusionPipeline.from_pretrained('stabilityai/stable-diffusion-2', scheduler=scheduler)#, safety_checker=None, torch_dtype=torch.float16)
         if is_cuda:
             self.pipe = self.pipe.to("cuda")
         self.resize_transform = transforms.Resize(512, interpolation=transforms.InterpolationMode.BILINEAR)
@@ -140,7 +141,6 @@ Weight Modulation for User Attribution and Fingerprinting in Text-to-Image Diffu
 
     gr.Markdown(
         """<h3>Demo: Text-to-Image (Stable diffusion 2) with random user attribution</h3>
-            WOUAF can be applied to other applications such as In-painting, Image-editing, Image Super-Resolution etc.
             <br>More details at: <a href="https://arxiv.org/abs/2306.04744">Paper</a>
         """
     )
@@ -171,7 +171,7 @@ Weight Modulation for User Attribution and Fingerprinting in Text-to-Image Diffu
             )
 
     with gr.Row():
-        steps = gr.Slider(label="Steps", minimum=1, maximum=50, value=45, step=1)
+        steps = gr.Slider(label="Steps", minimum=45, maximum=55, value=50, step=1)
         guidance_scale = gr.Slider(
             label="Guidance Scale", minimum=0, maximum=10, value=9, step=0.1
         )