update x

app.py

@@ -4,11 +4,14 @@ import gradio as gr
 import torch
 from diffusers import DiffusionPipeline
 
-inCloud = true
+import platform
+
+inCloud = True
 
 if inCloud:
     pipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
-    #pipe = pipe.to("mps")
+    if platform.system() == "Darwin": #Apple OS
+        pipe = pipe.to("mps")
     #pipe = pipe.to("cuda")
 
     # Recommended if your computer has < 64 GB of RAM
@@ -22,19 +25,30 @@ 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 generate_text_array(text):
-    text_array = text.split('\n\n')
+def generate_texts_array(text, max_length = 70):
+    text_array = []
+    start = 0
+
+    while start < len(text):
+        end = start + max_length
+        part = text[start:end]
+        text_array.append(part)
+        start = end
+
     return text_array
 
-def generation(text_array):
+def generation(text):
     images = []
+    text_array = generate_texts_array(text)
     images.append(pipe(text_array[0]).images[0])
-    images.append(pipe(text_array[0]).images[0])
-    images.append(pipe(text_array[0]).images[0])
+    images.append(pipe(text_array[1]).images[0])
+    images.append(pipe(text_array[2]).images[0])
     return images
 
 def generation_test(img):
@@ -46,8 +60,8 @@ def generation_test(img):
 
 demo = gr.Blocks()
 
-title = '# 3D print '
-description = 'App for detect errors in the 3D printing'
+title = '# Generando historias graficas '
+description = 'Escribe un texto y se generara una secuencia de imagenes basadas en el texto proporcionado'
 
 with demo:
     gr.Markdown(title)
@@ -66,10 +80,8 @@ with demo:
                 examples=textos,
                 inputs=text_input,
                 #outputs=im_2,
-
             )
             
-
         
     with gr.Row():
         imgs_output = []