Update interface_connection.py

interface_connection.py

@@ -7,7 +7,8 @@ import json
 
 device =torch.device('cpu')
 model_name_map = {
-    "wizard_tb 0.3": "demo_model_1"
+    "wizard_tb 0.3": "demo_model_1",
+    "wizard 0.3": "demo_model_0"
 }
 data_map = {
     "aBN_01": 0, "aBN_02": 1,
@@ -43,7 +44,7 @@ def text_out(h_mat, s_mat,model_name, data_name ):
 
 def compute_mat(data_name, model_name):
     print("model_name",model_name)
-    if model_name[0] == "wizard 0.3":
+    if model_name[0] == "wizard_tb 0.3":
         model = HGnn(edge_shape=51,
                      node_shape=2,
                      u_shape=10,