Update app.py

app.py

@@ -31,19 +31,19 @@ def generate_story_custom_category(texto,longitud=250,categoria='superheroe'):
   output = model_category.generate(input_ids, max_length=longitud,do_sample=True,repetition_penalty=1.1, temperature=1.2, top_p=0.95, top_k=50)
   return tokenizer_category.decode(output[0])
 
-contexto = gr.inputs.Textbox(lines=5, placeholder="Ingresar palabras claves para generar un cuento")
+contexto = gr.inputs.Textbox(lines=10, placeholder="Ingresar palabras claves para generar un cuento")
 categoria = gr.inputs.Textbox(lines=1, placeholder="Ingresar Categoria")
 longitud = gr.inputs.Slider(50, 500)
-opciones = gr.inputs.CheckboxGroup(["Generar", "Generar por Categoria", "Generar V 2.0"])
+opciones = gr.inputs.CheckboxGroup(["Generar", "Generar por Categoria", "Generar 2"])
 resultado = gr.outputs.HTML(label="Resultado")
-def generate_storie(contexto,opciones,categoria,longitud):
+def generate_storie(contexto,categoria,longitud,opciones):
   resultado=""
   cuentos=[]
   if "Generar" in opciones:
     cuentos.append(generate_story_custom(contexto,longitud))
   if "Generar por Categoria" in opciones:
     cuentos.append(generate_story_custom_category(contexto,longitud,categoria))
-  if "Generar V 2.0" in opciones:
+  if "Generar 2" in opciones:
     cuentos.append(generate_story_translate(contexto,longitud,categoria))  
   resultado += "<p><b>Generados:</b> "+'<b>Cuento: <b/>'.join(cuentos)+"</p>" 
   return resultado