Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -23,8 +23,8 @@ input_dims, output_dims = compute_input_output_dims(data_loaders['train'], extra
|
|
| 23 |
|
| 24 |
sampling_metrics = CrossDomainSamplingMetrics(data_loaders)
|
| 25 |
|
| 26 |
-
model = LGGMText2Graph_Demo.load_from_checkpoint('cc-deg.ckpt')
|
| 27 |
-
|
| 28 |
|
| 29 |
model.init_prompt_encoder_pretrained()
|
| 30 |
|
|
|
|
| 23 |
|
| 24 |
sampling_metrics = CrossDomainSamplingMetrics(data_loaders)
|
| 25 |
|
| 26 |
+
# model = LGGMText2Graph_Demo.load_from_checkpoint('cc-deg.ckpt', map_location=torch.device('cpu'))
|
| 27 |
+
model = LGGMText2Graph_Demo.load_from_checkpoint('cc-deg.ckpt', map_location=torch.device("cpu"))
|
| 28 |
|
| 29 |
model.init_prompt_encoder_pretrained()
|
| 30 |
|