update

app.py

@@ -1,5 +1,6 @@
 
 import gradio as gr
+#from transformers import pipeline
 
 import torch
 from diffusers import DiffusionPipeline
@@ -8,6 +9,9 @@ import platform
 
 inCloud = True
 
+#translator = pipeline("translation", model = "Helsinki-NLP/opus-mt-es-en")
+pipe = ''
+
 if inCloud:
     pipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
     if platform.system() == "Darwin": #Apple OS
@@ -25,11 +29,8 @@ Una noche, el ladrón desenterró el oro y se lo llevó. Cuando el rico descubri
 —Aquí tiene su tesoro. Sabe que nunca habría gastado sus lingotes. ¿Qué más le da, entonces, que sean piedras? Así por lo menos dejará de sufrir.'''
 ]
 
-
-
-def translate(text):
-    new_text = text
-    return new_text
+#def translate(text):
+#    return translator(text)[0]["translation_text"]
 
 def generate_texts_array(text, max_length = 70):
     text_array = []
@@ -45,10 +46,14 @@ def generate_texts_array(text, max_length = 70):
 
 def generation(text):
     images = []
+    steps = 5
+    #translate_text = translate(text)
     text_array = generate_texts_array(text)
-    images.append(pipe(text_array[0]).images[0])
-    images.append(pipe(text_array[1]).images[0])
-    images.append(pipe(text_array[2]).images[0])
+    images.append(pipe(text_array[0],num_inference_steps=steps).images[0])
+    images.append(pipe(text_array[1],num_inference_steps=steps).images[0])
+    images.append(pipe(text_array[2],num_inference_steps=steps).images[0])
+    images.append(pipe(text_array[3],num_inference_steps=steps).images[0])
+    images.append(pipe(text_array[4],num_inference_steps=steps).images[0])
     return images
 
 def generation_test(img):
@@ -56,6 +61,8 @@ def generation_test(img):
     images.append(img)
     images.append(img)
     images.append(img)
+    images.append(img)
+    images.append(img)
     return images
 
 demo = gr.Blocks()
@@ -73,6 +80,7 @@ with demo:
                                     info="Escribe una historia",
                                     lines=5, 
                                     value='Erase una vez..')
+            #steps_input = gr.Number(value=20)
             #img_input = gr.Image (type='pil')
             button = gr.Button(value="Generate")
 
@@ -88,6 +96,8 @@ with demo:
         imgs_output.append(gr.Image())
         imgs_output.append(gr.Image())
         imgs_output.append(gr.Image())
+        imgs_output.append(gr.Image())
+        imgs_output.append(gr.Image())
     
     button.click(generation, inputs=text_input, outputs=imgs_output)