Spaces:

akdeniz27
/

turkish-qna-with-xlm-roberta

Running

akdeniz27 commited on Nov 9, 2021

Commit

4917504

1 Parent(s): 36e84a7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -65,7 +65,9 @@ if Run_Button == True:
     qna_pipeline = setModel(model_checkpoint)
     output = qna_pipeline(question=selected_question, context=selected_context)
-    df = pd.DataFrame.from_dict(output, index=[0])
     cols_to_keep = ['answer','score','start','end']
     df_final = df[cols_to_keep]

     qna_pipeline = setModel(model_checkpoint)
     output = qna_pipeline(question=selected_question, context=selected_context)
+    output["answer"] = str(output["answer"])
+    df = pd.DataFrame.from_dict(output)
     cols_to_keep = ['answer','score','start','end']
     df_final = df[cols_to_keep]