Update classification.py

classification.py

@@ -207,6 +207,6 @@ def classification(column, file_path, categories, treshold):
     # Save data
     return save_data(df,file_path), df
 
-download_cate(cate_df):
+def download_cate(cate_df):
     cate_df.to_excel('categories.xlsx')
     return gr.File(value='categories.xlsx', visible=True)