Spaces:

NCEE-Build-Lab
/

watsonx.ai_Vector_Embedding_Visualizer_MNB

Running

App Files Files Community

MilanM commited on Apr 21

Commit

7dbf908

·

verified ·

1 Parent(s): 1bec912

Update visualizer_app.py

Files changed (1) hide show

visualizer_app.py +8 -3

visualizer_app.py CHANGED Viewed

@@ -528,9 +528,14 @@ def emb_model_state(get_embedding_model_list):
 @app.cell
 def emb_model_setup(embedding_model):
-    emb_model = embedding_model.value[0]['model_id']
-    emb_model_max_tk = embedding_model.value[0]['max_tokens']
-    emb_model_emb_dim = embedding_model.value[0]['embedding_dimensions']
     return emb_model, emb_model_emb_dim, emb_model_max_tk

 @app.cell
 def emb_model_setup(embedding_model):
+    if embedding_model is not None:
+        emb_model = embedding_model.value[0]['model_id']
+        emb_model_max_tk = embedding_model.value[0]['max_tokens']
+        emb_model_emb_dim = embedding_model.value[0]['embedding_dimensions']
+    else:
+        emb_model = None
+        emb_model_max_tk = None
+        emb_model_emb_dim = None
     return emb_model, emb_model_emb_dim, emb_model_max_tk