sps44's picture
added issue load
3c68a02
raw
history blame
728 Bytes
import os
import pandas as pd
import pickle
from renumics import spotlight
if __name__ == "__main__":
cache_file = "dataset_cache.parquet"
issue_cache_file="sliceline.pkl"
import pandas as pd
#load dataframe from parquet
df = pd.read_parquet(cache_file)
df_show = df.drop(columns=['embedding', 'probabilities'])
with open(issue_cache_file, "rb") as file:
issues = pickle.load(file)
while True:
view = spotlight.show(df_show, port=7860, host="0.0.0.0", issues=issues, layout="sliceline-layout.json",
dtype={"image": spotlight.Image, "embedding_reduced": spotlight.Embedding}, allow_filebrowsing=False)
view.close()