alitavanaali commited on
Commit
c0de39e
·
1 Parent(s): 0f298e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -280,8 +280,8 @@ def createDataframe(predictions, words):
280
  result1, result2 = zip(*sorted(zip(predictions, words), key=lambda x: x[0]))
281
  d = {'key':result1,'value':result2}
282
  df = pd.DataFrame(d)
283
- df1 = df[df['key'] != 'lyrics']
284
- df2 = df[df['key'] == 'lyrics']
285
  return df1, df2
286
 
287
  def isInside(w, z):
 
280
  result1, result2 = zip(*sorted(zip(predictions, words), key=lambda x: x[0]))
281
  d = {'key':result1,'value':result2}
282
  df = pd.DataFrame(d)
283
+ df1 = df.loc[df['key'] != 'lyrics']
284
+ df2 = df.loc[df['key'] == 'lyrics']
285
  return df1, df2
286
 
287
  def isInside(w, z):