Spaces:

yeq6x
/

TripletGeoEncoder-demo

Sleeping

yeq6x commited on Nov 13, 2024

Commit

2cf2a67

1 Parent(s): 26ba1d3

gpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -149,7 +149,7 @@ for model_info in models_info:
     feature_dim = model_info["feature_dim"]
     model_path = f"checkpoints/{model_name}"
     model = load_model(model_path, feature_dim)
-    models.append(model_name)
 setup(models_info[0])

     feature_dim = model_info["feature_dim"]
     model_path = f"checkpoints/{model_name}"
     model = load_model(model_path, feature_dim)
+    models.append(model)
 setup(models_info[0])