lily-hust commited on
Commit
1818d34
1 Parent(s): 018eb81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -31,9 +31,10 @@ def generate_df():
31
 
32
  @st.cache_data()
33
  def write_df(df, file, cls):
34
- rec = {'Image file name': file,
35
  'Class name': cls}
36
- df.append(rec, ignore_index = True)
 
37
 
38
  @st.cache_data()
39
  def convert_df(df):
 
31
 
32
  @st.cache_data()
33
  def write_df(df, file, cls):
34
+ rec = {'Image file name': file.name,
35
  'Class name': cls}
36
+ df = pd.concat([df, pd.DataFrame([rec])], ignore_index=True)
37
+ return df
38
 
39
  @st.cache_data()
40
  def convert_df(df):