Update app.py

app.py

@@ -35,7 +35,7 @@ pipe = DiffusionPipeline.from_pretrained(
     ),
     tokenizer=tokenizer,
     feature_extractor=feature_extractor,
-    torch_dtype=torch.float32,   # load weights in half-precision
+    torch_dtype=torch.bfloat16,   # load weights in half-precision
     use_safetensors=True
 )