strongpear
commited on
Commit
•
7bbedb1
1
Parent(s):
f772b19
update app.py
Browse files
app.py
CHANGED
@@ -258,7 +258,7 @@ class inferSSCL():
|
|
258 |
parser = argparse.ArgumentParser()
|
259 |
parser.add_argument('--task', default='infer')
|
260 |
parser.add_argument('--smooth_factor', type=float, default=0.9)
|
261 |
-
device = '
|
262 |
emb_size = 256
|
263 |
|
264 |
args = parser.parse_args(args=[])
|
|
|
258 |
parser = argparse.ArgumentParser()
|
259 |
parser.add_argument('--task', default='infer')
|
260 |
parser.add_argument('--smooth_factor', type=float, default=0.9)
|
261 |
+
device = 'cpu'
|
262 |
emb_size = 256
|
263 |
|
264 |
args = parser.parse_args(args=[])
|