a-v-bely commited on
Commit
b956367
·
verified ·
1 Parent(s): 494ed08

Update utilities_language_general/rus_constants.py

Browse files
utilities_language_general/rus_constants.py CHANGED
@@ -7,7 +7,7 @@ from pickle import load
7
  from transformers import pipeline
8
  from summarizer import Summarizer
9
  from torch import cuda, device
10
- device = device('cuda' if cuda.is_available else 'cpu')
11
 
12
 
13
  @st.cache_resource
 
7
  from transformers import pipeline
8
  from summarizer import Summarizer
9
  from torch import cuda, device
10
+ device = device('cpu')
11
 
12
 
13
  @st.cache_resource