fix inputs

app.py

@@ -11,43 +11,43 @@ login(token=HF_TOKEN)
 
 generator = pipeline("text-generation", model="artificial-nerds/gpt2-finetune-miio-v0.1")
 
-def generate_prompt(row):
-  prompt = f'Ciclo: {row["Ciclo"]}\nCiclo_Max: {row["Ciclo_Max"]}\nnumero_creditos: {row["numero_creditos"]}\nNroDiasAtraso: {row["NroDiasAtraso"]}\nNroCuotaDias: {row["NroCuotaDias"]}\nDiasCancel: {row["DiasCancel"]}\nDiaRestPago: {row["DiaRestPago"]}\nPctPagoDias_Prom: {row["PctPagoDias_Prom"]}\nPctPagoDias: {row["PctPagoDias"]}\nTipo_Cancelacion: {row["Tipo_Cancelacion"]}\nRango_PagoPlazo: {row["Rango_PagoPlazo"]}\nNroCuotasC: {row["NroCuotasC"]}\nSemanaCancelacion: {row["SemanaCancelacion"]}\ndeposit: {row["deposit"]}\nESTADO_CIVIL: {row["ESTADO_CIVIL"]}\n\nA partir de la información anterior, indica si el registro es Bueno o Malo para un crédito:'
+def generate_prompt(Ciclo, Ciclo_Max, numero_creditos, NroDiasAtraso, NroCuotaDias, DiasCancel, DiaRestPago, PctPagoDias_Prom, PctPagoDias, Tipo_Cancelacion, Rango_PagoPlazo, NroCuotasC, SemanaCancelacion, deposit, ESTADO_CIVIL):
+  prompt = f'Ciclo: {Ciclo}\nCiclo_Max: {Ciclo_Max}\nnumero_creditos: {numero_creditos}\nNroDiasAtraso: {NroDiasAtraso}\nNroCuotaDias: {NroCuotaDias}\nDiasCancel: {DiasCancel}\nDiaRestPago: {DiaRestPago}\nPctPagoDias_Prom: {PctPagoDias_Prom}\nPctPagoDias: {PctPagoDias}\nTipo_Cancelacion: {Tipo_Cancelacion}\nRango_PagoPlazo: {Rango_PagoPlazo}\nNroCuotasC: {NroCuotasC}\nSemanaCancelacion: {SemanaCancelacion}\ndeposit: {deposit}\nESTADO_CIVIL: {ESTADO_CIVIL}\n\nA partir de la información anterior, indica si el registro es Bueno o Malo para un crédito:'
   return prompt
 
 def test(prompt):
     outputs = generator(prompt, max_length=768, num_return_sequences=1, do_sample=True, top_p=0.95, top_k=50)
-    return outputs[0]["generated_text"]
+    text = outputs[0]["generated_text"]
+    result = text.splitlines()[-1].split(":")[-1].strip()
+    return result
 
 description = """
 Test GPT2 Model
 """
 
-default_prompt = """Ciclo: 1
-Ciclo_Max: 1
-numero_creditos: 1
-NroDiasAtraso: 2
-NroCuotaDias: 28
-DiasCancel: 2
-DiaRestPago: -26
-PctPagoDias_Prom: 7,143
-PctPagoDias: 7,143
-Tipo_Cancelacion: No_Liquidado_Vig
-Rango_%PagoPlazo: e. No_Liq_vig
-NroCuotasC: 4
-SemanaCancelacion: 2
-deposit: 500
-ESTADO_CIVIL: Casado
-
-A partir de la información anterior, indica si el registro es Bueno o Malo para un crédito:"""
+examples = [["1", "1", "1", "2", "28", "2", "-26", "7,143", "7,143", "No_Liquidado_Vig", "e. No_Liq_vig", "4", "2", "500", "Casado" ]]
 iface = gr.Interface(fn=test,
                       title= "GPT2 Model",
                       description = description,
                       inputs=[
-                              gr.inputs.Textbox(lines=4, placeholder="Prompt", label='Prompt')
+                              gr.inputs.Text(lines=1, placeholder="1", label='Ciclo'),
+                              gr.inputs.Textbox(lines=1, placeholder="1", label='Ciclo_Max'),
+                              gr.inputs.Textbox(lines=1, placeholder="1", label='numero_creditos'),
+                              gr.inputs.Textbox(lines=1, placeholder="2", label='NroDiasAtraso'),
+                              gr.inputs.Textbox(lines=1, placeholder="28", label='NroCuotasDias'),
+                              gr.inputs.Text(lines=1, placeholder="2", label='DiasCancel'),
+                              gr.inputs.Text(lines=1, placeholder="-26", label='DiaRestPago'),
+                              gr.inputs.Text(lines=1, placeholder="7,143", label='PctPagoDias_Prom'),
+                              gr.inputs.Text(lines=1, placeholder="7,143", label='PctPagoDias'),
+                              gr.inputs.Text(lines=1, placeholder="No_Liquidado_Vig", label='Tipo_Cancelacion'),
+                              gr.inputs.Text(lines=1, placeholder="e. No_Liq_vig", label='Rango_%PagoPlazo'),
+                              gr.inputs.Text(lines=1, placeholder="4", label='NroCuotas'),
+                              gr.inputs.Text(lines=1, placeholder="2", label='SemanaCancelación'),
+                              gr.inputs.Text(lines=1, placeholder="500", label='deposit'),
+                              gr.inputs.Text(lines=1, placeholder="Casado", label='ESTADO_CIVIL'),
                               ],
                       outputs=[
-                               gr.outputs.Textbox(label="Tu poema"),
+                               gr.outputs.Textbox(label="Resultado"),
                               ],
                       examples= [default_prompt]
                      )