Update app.py
Browse files
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 = "
|
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
|