Spaces:

ML-INTA
/

ML-Image

Running

ML-INTA commited on Oct 3, 2023

Commit

157a590

•

1 Parent(s): d0660e0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -41,7 +41,7 @@ def ia_imagenes(modelo, prompt, prompt_negativo, uploaded_file, my_strength, my_
     if modelo == "Texto":
-            from diffusers import StableDiffusionPipeline
             import torch
             pipe = StableDiffusionPipeline.from_pretrained(
@@ -61,7 +61,7 @@ def ia_imagenes(modelo, prompt, prompt_negativo, uploaded_file, my_strength, my_
     elif modelo == "Imagen":
-        from diffusers import StableDiffusionImg2ImgPipeline
         from PIL import Image
         import torch

     if modelo == "Texto":
+            from my_diffusers import StableDiffusionPipeline
             import torch
             pipe = StableDiffusionPipeline.from_pretrained(
     elif modelo == "Imagen":
+        from my_diffusers import StableDiffusionImg2ImgPipeline
         from PIL import Image
         import torch