Update app.py

app.py

@@ -3,8 +3,8 @@ import torch
 from diffusers import DiffusionPipeline
 
 # Load the model with authentication
-pipe = DiffusionPipeline.from_pretrained("Shakker-Labs/AWPortrait-FL", torch_dtype=torch.bfloat16, use_auth_token=True)
-pipe.enable_model_cpu_offload()  # Save some VRAM by offloading the model to CPU
+pipe = DiffusionPipeline.from_pretrained("Shakker-Labs/AWPortrait-FL", torch_dtype=torch.float32, use_auth_token=True)
+pipe.to("cpu")   # Save some VRAM by offloading the model to CPU
 
 def generate_image(prompt):
     image = pipe(