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 my_diffusers import StableDiffusionPipeline
|
| 45 |
import torch
|
| 46 |
|
| 47 |
pipe = StableDiffusionPipeline.from_pretrained(
|
|
|
|
| 61 |
|
| 62 |
elif modelo == "Imagen":
|
| 63 |
|
| 64 |
+
from my_diffusers import StableDiffusionImg2ImgPipeline
|
| 65 |
from PIL import Image
|
| 66 |
import torch
|
| 67 |
|