seanpedrickcase commited on
Commit
9e84863
1 Parent(s): 88d81fa

Importing space package near start of app now to avoid issue with cuda being initialised before

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. funcs/topic_core_funcs.py +0 -1
app.py CHANGED
@@ -4,6 +4,7 @@ import gradio as gr
4
  import pandas as pd
5
  import numpy as np
6
  from datetime import datetime
 
7
 
8
  from funcs.topic_core_funcs import pre_clean, optimise_zero_shot, extract_topics, reduce_outliers, represent_topics, visualise_topics, save_as_pytorch_model, change_default_vis_col
9
  from funcs.helper_functions import initial_file_load, custom_regex_load, ensure_output_folder_exists, output_folder, get_connection_params, get_or_create_env_var
 
4
  import pandas as pd
5
  import numpy as np
6
  from datetime import datetime
7
+ import spaces
8
 
9
  from funcs.topic_core_funcs import pre_clean, optimise_zero_shot, extract_topics, reduce_outliers, represent_topics, visualise_topics, save_as_pytorch_model, change_default_vis_col
10
  from funcs.helper_functions import initial_file_load, custom_regex_load, ensure_output_folder_exists, output_folder, get_connection_params, get_or_create_env_var
funcs/topic_core_funcs.py CHANGED
@@ -7,7 +7,6 @@ import pandas as pd
7
  import numpy as np
8
  import time
9
  from bertopic import BERTopic
10
- import spaces
11
 
12
  from typing import List, Type, Union
13
  PandasDataFrame = Type[pd.DataFrame]
 
7
  import numpy as np
8
  import time
9
  from bertopic import BERTopic
 
10
 
11
  from typing import List, Type, Union
12
  PandasDataFrame = Type[pd.DataFrame]