Update app.py

app.py

@@ -20,16 +20,24 @@ model_map = {
              
              
              
-             'Long Hair' : 'models/longhair.pt',
+             'Long Hair' : 'models/long_hair.pt',
              'Curly Hair' : 'models/curlyhair.pt',
              
              'Pixar Style' : 'models/pixar_style.pt',
              'Sculpture Style': 'models/sculpture_style.pt',
+             'Clay Style': 'models/clay_style.pt',
              
              'Repair Images': 'models/repair_slider.pt',
              'Fix Hands': 'models/fix_hands.pt',
+
+             'Cluttered Room': 'models/cluttered_room.pt',
+
+             'Dark Weather': 'models/dark_weather.pt',
+             'Festive': 'models/festive.pt',
+             'Tropical Weather': 'models/tropical_weather.pt',
+             'Winter Weather': 'models/winter_weather.pt',
              
-             'Wavy Eyebrows': 'models/eyebrows.pt',
+             'Wavy Eyebrows': 'models/eyebrow.pt',
              'Small Eyes (use scales -3, -1, 1, 3)': 'models/eyesize.pt',
             }
 
@@ -48,12 +56,12 @@ class Demo:
         self.training = False
         self.generating = False
         self.device = 'cuda'
-        self.weight_dtype = torch.float16
+        self.weight_dtype = torch.bfloat16
         self.pipe = StableDiffusionXLPipeline.from_pretrained('stabilityai/stable-diffusion-xl-base-1.0', torch_dtype=self.weight_dtype).to(self.device)
         self.pipe.enable_xformers_memory_efficient_attention()
         with gr.Blocks() as demo:
             self.layout()
-            demo.queue(max_size=5).launch(share=True, max_threads=3)
+            demo.queue(max_size=5).launch(share=True, max_threads=2)
 
 
     def layout(self):
@@ -98,8 +106,8 @@ class Demo:
                             )
                             
                             self.slider_scale_infr = gr.Slider(
-                                -6,
-                                6,
+                                -4,
+                                4,
                                 label="Slider Scale",
                                 value=3,
                                 info="Larger slider scale result in stronger edit"