File size: 521 Bytes
5bcc73a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
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(),
}
)
|