ai-forever commited on
Commit
629af13
·
1 Parent(s): 3221dbb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ article = (
20
  "</p>"
21
  )
22
 
23
- device = "cuda" if torch.cuda.is_available() else "cpu"
24
  fp16 = device != 'cpu'
25
 
26
  def transl(text, src='en', dest='hy' ):
 
20
  "</p>"
21
  )
22
 
23
+ device = "cuda:0" if torch.cuda.is_available() else "cpu"
24
  fp16 = device != 'cpu'
25
 
26
  def transl(text, src='en', dest='hy' ):