Hugging Face's logo

Spaces:
YuWang0103
/
LGGM-Text2Graph
Runtime error

App Files Community
YuWang0103 commited on
Commit
0a172b6
·
verified ·
1 Parent(s): 719a04f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- # model = LGGMText2Graph_Demo.load_from_checkpoint('cc-deg.ckpt', map_location=torch.device("cpu"))
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