Update app.py

app.py

@@ -27,7 +27,9 @@ device = torch.device(
     "cuda" if torch.cuda.is_available() else "xpu" if xpu_available else "cpu"
 )
 torch_device = device
-torch_dtype = torch.float16
+
+#torch_dtype = torch.float16
+torch_dtype = torch.bfloat16
 
 print(f"SAFETY_CHECKER: {SAFETY_CHECKER}")
 print(f"device: {device}")