Karpernik commited on
Commit
3c4bffd
·
verified ·
1 Parent(s): 7451219

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ st.write('Данный сервис предназначен для выбора
13
  'Поддерживается ввод только английского языка.')
14
  st.markdown('#### Введите название статьи и ее краткое содержание:')
15
 
16
- device = torch.device('cuda:0') if torch.cuda.is_available() else torch.device('cpu')
17
 
18
  def create_model_and_optimizer(model, lr=1e-4, beta1=0.9, beta2=0.999, device=device):
19
  model = model.to(device)
 
13
  'Поддерживается ввод только английского языка.')
14
  st.markdown('#### Введите название статьи и ее краткое содержание:')
15
 
16
+ device = torch.device('cpu')
17
 
18
  def create_model_and_optimizer(model, lr=1e-4, beta1=0.9, beta2=0.999, device=device):
19
  model = model.to(device)