Spaces:
Runtime error
Runtime error
liuyizhang
commited on
Commit
·
af76788
1
Parent(s):
9988241
update app.py
Browse files
app.py
CHANGED
|
@@ -347,8 +347,8 @@ config = dict(
|
|
| 347 |
)
|
| 348 |
config = Config(config)
|
| 349 |
|
| 350 |
-
class Ram_Predictor(RamPredictor
|
| 351 |
-
def __init__(self,config):
|
| 352 |
self.config = config
|
| 353 |
self.device = torch.device(device)
|
| 354 |
self._build_model()
|
|
|
|
| 347 |
)
|
| 348 |
config = Config(config)
|
| 349 |
|
| 350 |
+
class Ram_Predictor(RamPredictor):
|
| 351 |
+
def __init__(self, config, device='cpu'):
|
| 352 |
self.config = config
|
| 353 |
self.device = torch.device(device)
|
| 354 |
self._build_model()
|