MostafaAly commited on
Commit
76958e7
·
verified ·
1 Parent(s): a353928

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ logging.set_verbosity_error() # This suppresses warnings, including cache migra
8
  from diffusers import StableDiffusionPipeline
9
 
10
  # Load the model
11
- model_id = "MostafaAly/stable-diffusion-finetuned"
12
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
13
 
14
  # Check if CUDA is available and move the model to GPU if it is, otherwise use CPU
 
8
  from diffusers import StableDiffusionPipeline
9
 
10
  # Load the model
11
+ model_id = "ImageInception/stable-diffusion-finetuned"
12
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
13
 
14
  # Check if CUDA is available and move the model to GPU if it is, otherwise use CPU