adri017 commited on
Commit
2b0cb07
·
1 Parent(s): ddbdab0

ultima prueba

Browse files
Files changed (1) hide show
  1. appDiffuserNegocio.py +3 -3
appDiffuserNegocio.py CHANGED
@@ -1,10 +1,10 @@
1
  import gradio as gr
2
  import torch
3
- from diffusers import StableDiffusionPipeline
4
- pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-v1-5")
5
 
6
  def generarImagen(texto):
7
- imagen = pipe(prompt=texto).images[0]
8
  return imagen
9
 
10
  with gr.Blocks() as demo:
 
1
  import gradio as gr
2
  import torch
3
+ from transformers import pipeline
4
+ modelo = pipeline("text-to-image", model="runwayml/stable-diffusion-v1-5")
5
 
6
  def generarImagen(texto):
7
+ imagen = modelo(prompt=texto).images[0]
8
  return imagen
9
 
10
  with gr.Blocks() as demo: