akdeniz27 commited on
Commit
36e84a7
·
1 Parent(s): b074416

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ if Run_Button == True:
65
  qna_pipeline = setModel(model_checkpoint)
66
  output = qna_pipeline(question=selected_question, context=selected_context)
67
 
68
- df = pd.DataFrame.from_dict(output)
69
  cols_to_keep = ['answer','score','start','end']
70
  df_final = df[cols_to_keep]
71
 
 
65
  qna_pipeline = setModel(model_checkpoint)
66
  output = qna_pipeline(question=selected_question, context=selected_context)
67
 
68
+ df = pd.DataFrame.from_dict(output, index=[0])
69
  cols_to_keep = ['answer','score','start','end']
70
  df_final = df[cols_to_keep]
71