save_image.py

from hashlib import sha256
import streamlit as st
from datetime import datetime, timezone
from scheduler import ParquetScheduler


# scheduler = ParquetScheduler(
#     repo_id=st.secrets.IMAGES_REPO,
#     token=st.secrets.HF_TOKEN,
#     every=15,
# )
#
#
# @st.cache_data(ttl=3600)
# def save_image(data, prediction):
#     scheduler.append(
#         {
#             "date": datetime.now(timezone.utc).date(),
#             "image": data,
#             "prediction": prediction,
#             "sha256": sha256(data).hexdigest(),
#         }
#     )