Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -31,6 +31,20 @@ if device == 'cuda:0':
|
|
31 |
pipe = pipeline("question-answering",model_ans,tokenizer =tokenizer_ans,device = 0)
|
32 |
else:
|
33 |
pipe = pipeline("question-answering",model_ans,tokenizer =tokenizer_ans)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
34 |
|
35 |
def cls_pooling(model_output):
|
36 |
return model_output.last_hidden_state[:,0]
|
@@ -203,21 +217,21 @@ with gr.Blocks() as demo:
|
|
203 |
Adjunte su documento, ya sea en formato .txt o .pdf, y pregunte lo que desee.
|
204 |
|
205 |
""")
|
206 |
-
|
207 |
-
label="Sus documentos subidos (PDF o txt)"
|
208 |
-
dataset = gr.Dataframe(
|
209 |
-
|
210 |
-
|
211 |
-
|
212 |
-
|
213 |
-
|
214 |
-
)
|
215 |
-
buildb = gr.Textbox("⚠️Esperando documentos...",
|
216 |
-
|
217 |
-
dataset.change(validate_dataset, inputs=[
|
218 |
-
|
219 |
-
uploaded_files.change(request_pathname, inputs=[
|
220 |
-
|
221 |
query = gr.Textbox(
|
222 |
placeholder="Introduzca su pregunta aquí...", label="Pregunta")
|
223 |
ask = gr.Button("Preguntar")
|
@@ -230,7 +244,7 @@ with gr.Blocks() as demo:
|
|
230 |
# ask.click(fn=do_ask, inputs=[query, buildb,
|
231 |
# dataset], outputs=[answer, context])
|
232 |
ask.click(fn=predict, inputs=[query,
|
233 |
-
|
234 |
|
235 |
demo.queue(concurrency_count=20)
|
236 |
demo.launch(show_error=True)
|
|
|
31 |
pipe = pipeline("question-answering",model_ans,tokenizer =tokenizer_ans,device = 0)
|
32 |
else:
|
33 |
pipe = pipeline("question-answering",model_ans,tokenizer =tokenizer_ans)
|
34 |
+
|
35 |
+
def validate_dataset(dataset):
|
36 |
+
global docs
|
37 |
+
docs = None # clear it out if dataset is modified
|
38 |
+
docs_ready = dataset.iloc[-1, 0] != ""
|
39 |
+
if docs_ready:
|
40 |
+
return "✨Listo✨"
|
41 |
+
else:
|
42 |
+
return "⚠️Esperando documentos..."
|
43 |
+
|
44 |
+
def request_pathname(files):
|
45 |
+
if files is None:
|
46 |
+
return [[]]
|
47 |
+
return [[file.name, file.name.split('/')[-1]] for file in files]
|
48 |
|
49 |
def cls_pooling(model_output):
|
50 |
return model_output.last_hidden_state[:,0]
|
|
|
217 |
Adjunte su documento, ya sea en formato .txt o .pdf, y pregunte lo que desee.
|
218 |
|
219 |
""")
|
220 |
+
file = gr.File(
|
221 |
+
label="Sus documentos subidos (PDF o txt)")
|
222 |
+
# dataset = gr.Dataframe(
|
223 |
+
# headers=["filepath", "citation string"],
|
224 |
+
# datatype=["str", "str"],
|
225 |
+
# col_count=(2, "fixed"),
|
226 |
+
# interactive=True,
|
227 |
+
# label="Documentos y citas"
|
228 |
+
# )
|
229 |
+
# buildb = gr.Textbox("⚠️Esperando documentos...",
|
230 |
+
# label="Estado", interactive=False, show_label=True)
|
231 |
+
# dataset.change(validate_dataset, inputs=[
|
232 |
+
# dataset], outputs=[buildb])
|
233 |
+
# uploaded_files.change(request_pathname, inputs=[
|
234 |
+
# uploaded_files], outputs=[dataset])
|
235 |
query = gr.Textbox(
|
236 |
placeholder="Introduzca su pregunta aquí...", label="Pregunta")
|
237 |
ask = gr.Button("Preguntar")
|
|
|
244 |
# ask.click(fn=do_ask, inputs=[query, buildb,
|
245 |
# dataset], outputs=[answer, context])
|
246 |
ask.click(fn=predict, inputs=[query,
|
247 |
+
file], outputs=[answer])
|
248 |
|
249 |
demo.queue(concurrency_count=20)
|
250 |
demo.launch(show_error=True)
|