sps44's picture
prints added
3afaf74
raw
history blame
760 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"
print('starting up')
#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)
print('issues loaded')
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()