MilanM commited on
Commit
7e99f82
·
verified ·
1 Parent(s): 8816b1d

Update visualizer_app.py

Browse files
Files changed (1) hide show
  1. visualizer_app.py +23 -10
visualizer_app.py CHANGED
@@ -140,10 +140,16 @@ def client_instantiation(
140
  else:
141
  deployment_client = None
142
 
 
143
  if project_client is not None:
144
  task_credentials_details = setup_task_credentials(project_client)
145
- else:
146
  task_credentials_details = setup_task_credentials(deployment_client)
 
 
 
 
 
147
  else:
148
  wx_credentials = None
149
  project_client = None
@@ -433,7 +439,7 @@ def active_client(client_selector):
433
 
434
  @app.cell
435
  def emb_model_selection(client, set_embedding_model_list):
436
- if client is not None:
437
  model_specs = client.foundation_models.get_embeddings_model_specs()
438
  # model_specs = client.foundation_models.get_model_specs()
439
  resources = model_specs["resources"]
@@ -509,16 +515,22 @@ def embedding_model():
509
 
510
  @app.cell
511
  def emb_model_parameters(emb_model_max_tk):
512
- from ibm_watsonx_ai.foundation_models import Embeddings
513
  from ibm_watsonx_ai.metanames import EmbedTextParamsMetaNames as EmbedParams
514
-
515
- embed_params = {
516
- EmbedParams.TRUNCATE_INPUT_TOKENS: emb_model_max_tk,
517
- EmbedParams.RETURN_OPTIONS: {
518
- 'input_text': True
 
519
  }
520
- }
521
- return Embeddings, embed_params
 
 
 
 
 
 
522
 
523
 
524
  @app.cell
@@ -542,6 +554,7 @@ def emb_model_setup(embedding_model):
542
 
543
  @app.cell
544
  def emb_model_instantiation(Embeddings, client, emb_model, embed_params):
 
545
  if client is not None:
546
  embedding = Embeddings(
547
  model_id=emb_model,
 
140
  else:
141
  deployment_client = None
142
 
143
+
144
  if project_client is not None:
145
  task_credentials_details = setup_task_credentials(project_client)
146
+ elif deployment_client is not None:
147
  task_credentials_details = setup_task_credentials(deployment_client)
148
+ elif project_client is not None and deployment_client is not None:
149
+ task_credentials_details = setup_task_credentials(project_client)
150
+ else:
151
+ task_credentials_details = None
152
+
153
  else:
154
  wx_credentials = None
155
  project_client = None
 
439
 
440
  @app.cell
441
  def emb_model_selection(client, set_embedding_model_list):
442
+ if client_instantiation_form.value and client is not None:
443
  model_specs = client.foundation_models.get_embeddings_model_specs()
444
  # model_specs = client.foundation_models.get_model_specs()
445
  resources = model_specs["resources"]
 
515
 
516
  @app.cell
517
  def emb_model_parameters(emb_model_max_tk):
 
518
  from ibm_watsonx_ai.metanames import EmbedTextParamsMetaNames as EmbedParams
519
+ if client is not None:
520
+ embed_params = {
521
+ EmbedParams.TRUNCATE_INPUT_TOKENS: emb_model_max_tk,
522
+ EmbedParams.RETURN_OPTIONS: {
523
+ 'input_text': True
524
+ }
525
  }
526
+ else:
527
+ embed_params = {
528
+ EmbedParams.TRUNCATE_INPUT_TOKENS: 128,
529
+ EmbedParams.RETURN_OPTIONS: {
530
+ 'input_text': True
531
+ }
532
+ }
533
+ return embed_params
534
 
535
 
536
  @app.cell
 
554
 
555
  @app.cell
556
  def emb_model_instantiation(Embeddings, client, emb_model, embed_params):
557
+ from ibm_watsonx_ai.foundation_models import Embeddings
558
  if client is not None:
559
  embedding = Embeddings(
560
  model_id=emb_model,