vardaan123 commited on
Commit
2840071
·
1 Parent(s): 85f8c1c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ num_ent_id = len(entity2id)
31
  target_list = generate_target_list(datacsv, entity2id) # Assuming this function is defined elsewhere
32
 
33
  # Initialize your model here
34
- model = DistMult(args, num_ent_id, target_list, torch.device('cpu')) # Update arguments as necessary
35
  model.eval()
36
 
37
  # Define your evaluation function
 
31
  target_list = generate_target_list(datacsv, entity2id) # Assuming this function is defined elsewhere
32
 
33
  # Initialize your model here
34
+ model = DistMult(num_ent_id, target_list, torch.device('cpu')) # Update arguments as necessary
35
  model.eval()
36
 
37
  # Define your evaluation function