Update app.py

app.py

@@ -119,7 +119,7 @@ def load_and_prepare_model(model_id):
     )
     sched = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear", algorithm_type="sde-dpmsolver++")
     pipe.scheduler=sched
-    pipe.to(device=device)
+    pipe.to(device=device, dtype=torch.bfloat16)
     return pipe
 
 # Preload and compile both models
@@ -156,7 +156,7 @@ def generate_60(
     prompt: str,
     negative_prompt: str = "",
     use_negative_prompt: bool = False,
-  #  style_selection: str = DEFAULT_STYLE_NAME,
+    style_selection: str = "",
     seed: int = 1,
     width: int = 768,
     height: int = 768,
@@ -206,7 +206,7 @@ def generate_90(
     prompt: str,
     negative_prompt: str = "",
     use_negative_prompt: bool = False,
-   # style_selection: str = DEFAULT_STYLE_NAME,
+    style_selection: str = "",
     seed: int = 1,
     width: int = 768,
     height: int = 768,