Spaces:
Runtime error
Runtime error
ai-forever
commited on
Commit
·
629af13
1
Parent(s):
3221dbb
Update app.py
Browse files
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' ):
|