Spaces:
Runtime error
Runtime error
cocktailpeanut
commited on
Commit
•
6d72cb2
1
Parent(s):
127d24d
localize
Browse files
dialoggen/dialoggen_demo.py
CHANGED
@@ -29,7 +29,7 @@ import re
|
|
29 |
import devicetorch
|
30 |
|
31 |
|
32 |
-
device =
|
33 |
|
34 |
def image_parser(image_file, sep=','):
|
35 |
out = image_file.split(sep)
|
|
|
29 |
import devicetorch
|
30 |
|
31 |
|
32 |
+
device = devicetorch.get(torch)
|
33 |
|
34 |
def image_parser(image_file, sep=','):
|
35 |
out = image_file.split(sep)
|