SpawnedShoyo commited on
Commit
49a6865
·
verified ·
1 Parent(s): 3158c87

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,8 @@ from diffusers import DiffusionPipeline
4
  from PIL import Image, ImageDraw, ImageFont
5
 
6
  # Load the model
7
- model = DiffusionPipeline.from_pretrained("ginipick/flux-lora-eric-cat", torch_dtype=torch.float16)
8
- model = model.to("cuda") # Move the model to GPU if available
9
 
10
  def generate_image(caption):
11
  # Generate the image from the caption
 
4
  from PIL import Image, ImageDraw, ImageFont
5
 
6
  # Load the model
7
+ model = DiffusionPipeline.from_pretrained("ginipick/flux-lora-eric-cat", torch_dtype=torch.float32)
8
+ # No need to move the model to GPU since we're using CPU
9
 
10
  def generate_image(caption):
11
  # Generate the image from the caption