Upload 2 files

app.py

@@ -1,35 +1,23 @@
-from diffusers import StableDiffusionPipeline,  EulerDiscreteScheduler
-from transformers import pipeline
-import torch
 import gradio as gr
-
-model_name = "stabilityai/stable-diffusion-2"
-scheduler = EulerDiscreteScheduler.from_pretrained(model_name, subfolder="scheduler")
-pipe = StableDiffusionPipeline.from_pretrained(model_name, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)
-
-# Movemos el pipeline a GPU para tener una inferencia más rápida.
-device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = pipe.to(device)
-pipe.enable_attention_slicing() 
+from transformers import pipeline
 
 # Cargamos el modelo traductor y creamos el pipeline para traducir al ingles
 spanish_model_name = "Helsinki-NLP/opus-mt-es-en"
 translator_es_en = pipeline("translation", model=spanish_model_name)
 
-def predict(text):
+def translate(text):
   prompt_es = text
   english_text = translator_es_en(prompt_es)
   prompt_en = english_text[0]['translation_text']
-  image = pipe(prompt_en).images[0]
-  return image
+  return prompt_en
 
-description = """
-<h2 style="text-align:center">Programa para generar imágenes a partir de texto con Stable Diffusion 2</h2>
-<h2 style="text-align:center">Prompt en español!!</h2>
+title="Texto a Imagen en Español con Stable Diffusion 2.0"
+article = """
+El modelo usa: 
+- Texto a Imagen [Stable Diffusion 2.0](https://huggingface.co/stabilityai/stable-diffusion-2), 
+- Para las traducciones [Helsinki-NLP](https://huggingface.co/Helsinki-NLP) 
+\n ... y mucha magia ☺
 """
-
-gr.Interface(fn=predict,
-              title="Texto a Imagen en Español",
+gr.Interface(fn=translate,
               inputs= gr.Textbox("", max_lines = 2, label = "Inserte su texto aqui"),
-              outputs = "image",
-              description = description).launch(debug=True)
+              outputs = "text").load("models/stabilityai/stable-diffusion-2", title = title, article=article).launch(debug=True)

requirements.txt

@@ -2,7 +2,4 @@ transformers
 gradio
 torch
 sacremoses==0.0.53
-sentencepiece==0.1.97
-accelerate 
-scipy
-diffusers @ git+https://github.com/huggingface/diffusers.git 
+sentencepiece==0.1.97