friendshipkim commited on
Commit
56ae08e
·
1 Parent(s): d057d13

print model

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -56,6 +56,7 @@ def get_candidate_strings():
56
  @st.cache
57
  def get_embedding_dataset(model):
58
  # data = load_from_disk(f"{ASSETS_PATH}/{model}/embedding")
 
59
  data = load_dataset(f"{ORG_ID}/{model}_embedding", use_auth_token=TOKEN)
60
  return data
61
 
 
56
  @st.cache
57
  def get_embedding_dataset(model):
58
  # data = load_from_disk(f"{ASSETS_PATH}/{model}/embedding")
59
+ st.write(model)
60
  data = load_dataset(f"{ORG_ID}/{model}_embedding", use_auth_token=TOKEN)
61
  return data
62