Update app.py

app.py

@@ -280,8 +280,8 @@ def createDataframe(predictions, words):
     result1, result2 = zip(*sorted(zip(predictions, words), key=lambda x: x[0]))
     d = {'key':result1,'value':result2}
     df = pd.DataFrame(d)
-    df1 = df.loc[df['key'] != 'lyrics']
-    df2 = df.loc[df['key'] == 'lyrics']
+    df1 = df[df['key'] != 'lyrics']
+    df2 = df[df['key'] == 'lyrics']
     return df1, df2
 
 def isInside(w, z):
@@ -371,7 +371,7 @@ def completepreprocess(pdffile):
     print('page: ' + str(i) + '  ' + str(len(_preds))+ '  ' + str(len(_words)))
     df1, df2 = createDataframe(_preds, _words)
     dfOne=dfOne.append(df1)
-    dfTwo=dfOne.append(df2)
+    dfTwo=dfTwo.append(df2)
 
   im2=mergeImageVertical(a)  
   return im2,dfOne,dfTwo