Spaces:
Sleeping
Sleeping
Commit
·
2840071
1
Parent(s):
85f8c1c
Update app.py
Browse files
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(
|
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
|