Spaces:
Runtime error
Runtime error
Update model.py
Browse files
model.py
CHANGED
@@ -18,7 +18,7 @@ config = dict(
|
|
18 |
|
19 |
# model info
|
20 |
tokenizer_path = 'allenai/biomed_roberta_base', # 'roberta-base',
|
21 |
-
model_checkpoint = '
|
22 |
device = 'cuda' if torch.cuda.is_available() else 'cpu',
|
23 |
|
24 |
# training paramters
|
|
|
18 |
|
19 |
# model info
|
20 |
tokenizer_path = 'allenai/biomed_roberta_base', # 'roberta-base',
|
21 |
+
model_checkpoint = 'allenai/biomed_roberta_base', # 'roberta-base',
|
22 |
device = 'cuda' if torch.cuda.is_available() else 'cpu',
|
23 |
|
24 |
# training paramters
|