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()
|