process.py

import csv
import os
from huggingface_hub import Repository
from datetime import datetime
from torchvision.utils import save_image

DATASET_REPO_URL = "https://huggingface.co/datasets/bluelu/bgi"
DATA_FILENAME = "sc.csv"
DATA_FILE = os.path.join("data", DATA_FILENAME)
IN = os.path.join("data", str(datetime.now().replace(microsecond=0)) + 'sc.png')
HF_TOKEN = os.environ.get("S2")


def fiximg(imgs):
    repo = Repository(
        local_dir="data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN)

    save_image(imgs, IN)


    with open(DATA_FILE, 'a') as file:
        writer = csv.DictWriter(file, fieldnames=["ime"])
        writer.writerow({"ime": str(datetime.now())})
    repo.push_to_hub()