Spaces:
Runtime error
Runtime error
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() | |