Update app.py

app.py

@@ -10,9 +10,9 @@ os.environ["TOKENIZERS_PARALLELISM"] = "true"
 generator = pipeline('text-generation', model="salomonsky/deepSP")
 tokenizer = GPT2Tokenizer.from_pretrained('salomonsky/deepSP')
 
-def generate_output(text, image):
-    prompt = ""
-    input_tokens = tokenizer.encode(text, add_special_tokens=False)
+def generate_output(name, date_of_birth, image):
+    prompt = f"Mi nombre es Andrea. Soy un chatbot de IA. Mucho gusto, {name}.\nNací el {date_of_birth}."
+    input_tokens = tokenizer.encode(prompt, add_special_tokens=False)
     input_text = tokenizer.decode(input_tokens)
     gpt2_output = generator(input_text, max_length=20, do_sample=True, temperature=0.9)
     generated_text = gpt2_output[0]['generated_text']
@@ -45,15 +45,18 @@ thumbnail_width = "50px"
 iface = gr.Interface(
     fn=generate_output,
     inputs=[
-        gr.inputs.Textbox(lines=1, placeholder='Escribe tu nombre'),
-        gr.inputs.Radio(choices, label="Selecciona una vidente:")
+        gr.inputs.Textbox(lines=1, label="Nombre", placeholder="Ingresa tu nombre"),
+        gr.inputs.Textbox(lines=1, label="Fecha de Nacimiento", placeholder="DD/MM/AAAA"),
+        gr.inputs.Radio(choices, label="Selecciona una imagen:")
     ],
     outputs=[
-        gr.outputs.Video(label="Respuesta (un minuto aproximadamente)"),
-        gr.outputs.Textbox(label="Mensaje de error", type="auto")
+        gr.outputs.Video(label="Respuesta de Andrea (un minuto aproximadamente)"),
+        gr.outputs.Textbox(label="Mensaje de error", type="text")
     ],
-    title="Oraculo con Inteligencia Artifical v.2(c)",
-    thumbnail_width=thumbnail_width
+    title="Oraculo de Inteligencia Artifical v2.1",
+    thumbnail_width=thumbnail_width,
+    layout="vertical",
+    description="Por favor, ingresa tu nombre y fecha de nacimiento."
 )
 
 iface.launch()