Update classification.py

classification.py

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