Update app.py

app.py

@@ -31,6 +31,20 @@ if device == 'cuda:0':
     pipe = pipeline("question-answering",model_ans,tokenizer =tokenizer_ans,device = 0)
 else:
     pipe = pipeline("question-answering",model_ans,tokenizer =tokenizer_ans)
+
+def validate_dataset(dataset):
+    global docs
+    docs = None  # clear it out if dataset is modified
+    docs_ready = dataset.iloc[-1, 0] != ""
+    if docs_ready:
+        return "✨Listo✨"
+    else:
+        return "⚠️Esperando documentos..."
+
+def request_pathname(files):
+    if files is None:
+        return [[]]
+    return [[file.name, file.name.split('/')[-1]] for file in files]
     
 def cls_pooling(model_output):
     return model_output.last_hidden_state[:,0]
@@ -203,21 +217,21 @@ with gr.Blocks() as demo:
     Adjunte su documento, ya sea en formato .txt o .pdf, y pregunte lo que desee.
     
     """)
-    uploaded_files = gr.File(
-        label="Sus documentos subidos (PDF o txt)", file_count="multiple", )
-    dataset = gr.Dataframe(
-        headers=["filepath", "citation string"],
-        datatype=["str", "str"],
-        col_count=(2, "fixed"),
-        interactive=True,
-        label="Documentos y citas"
-    )
-    buildb = gr.Textbox("⚠️Esperando documentos...",
-                        label="Estado", interactive=False, show_label=True)
-    dataset.change(validate_dataset, inputs=[
-                   dataset], outputs=[buildb])
-    uploaded_files.change(request_pathname, inputs=[
-                          uploaded_files], outputs=[dataset])
+    file = gr.File(
+        label="Sus documentos subidos (PDF o txt)")
+    # dataset = gr.Dataframe(
+    #     headers=["filepath", "citation string"],
+    #     datatype=["str", "str"],
+    #     col_count=(2, "fixed"),
+    #     interactive=True,
+    #     label="Documentos y citas"
+    # )
+    # buildb = gr.Textbox("⚠️Esperando documentos...",
+    #                     label="Estado", interactive=False, show_label=True)
+    # dataset.change(validate_dataset, inputs=[
+    #                dataset], outputs=[buildb])
+    # uploaded_files.change(request_pathname, inputs=[
+    #                       uploaded_files], outputs=[dataset])
     query = gr.Textbox(
         placeholder="Introduzca su pregunta aquí...", label="Pregunta")
     ask = gr.Button("Preguntar")
@@ -230,7 +244,7 @@ with gr.Blocks() as demo:
     # ask.click(fn=do_ask, inputs=[query, buildb,
     #                              dataset], outputs=[answer, context])
     ask.click(fn=predict, inputs=[query,
-                                 gr.inputs.File()], outputs=[answer])
+                                 file], outputs=[answer])
 
 demo.queue(concurrency_count=20)
 demo.launch(show_error=True)