Update app.py
Browse files
app.py
CHANGED
@@ -41,7 +41,7 @@ def ia_imagenes(modelo, prompt, prompt_negativo, uploaded_file, my_strength, my_
|
|
41 |
|
42 |
if modelo == "Texto":
|
43 |
|
44 |
-
from
|
45 |
import torch
|
46 |
|
47 |
pipe = StableDiffusionPipeline.from_pretrained(
|
@@ -61,7 +61,7 @@ def ia_imagenes(modelo, prompt, prompt_negativo, uploaded_file, my_strength, my_
|
|
61 |
|
62 |
elif modelo == "Imagen":
|
63 |
|
64 |
-
from
|
65 |
from PIL import Image
|
66 |
import torch
|
67 |
|
|
|
41 |
|
42 |
if modelo == "Texto":
|
43 |
|
44 |
+
from diffusers import StableDiffusionPipeline
|
45 |
import torch
|
46 |
|
47 |
pipe = StableDiffusionPipeline.from_pretrained(
|
|
|
61 |
|
62 |
elif modelo == "Imagen":
|
63 |
|
64 |
+
from diffusers import StableDiffusionImg2ImgPipeline
|
65 |
from PIL import Image
|
66 |
import torch
|
67 |
|