Spaces:
Runtime error
Runtime error
File size: 2,474 Bytes
af44b2b 54c0759 af44b2b 54c0759 af44b2b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
# https://huggingface.co/docs/huggingface_hub/guides/upload
# PUSH model to HuggingFace manufy/mnist_model_keras
# using upload_folder, commit message is set by defaulr and not customizable
###############################
# 1 - Initial generic imports #
###############################
from icecream import ic
ic("--- Importing generic libraries ---")
from dotenv import load_dotenv, find_dotenv
import os
###############################################################################
# 2 - Setting up environment variable HUGGINGFACE_TOKEN with write permission #
###############################################################################
ic("--- Importing HF API token ---")
load_dotenv(find_dotenv())
token = os.getenv("HUGGINGFACE_TOKEN")
# The code in the comments is not needed
# the model doen't need to be loaded to be uploaded
################################################################
# 3 - Importing keras load_model #
################################################################
# ic("--- Importing tensorflow load_model ---")
# from tensorflow.keras.models import load_model
# ic("--- Importing huggingface_hub ---")
###########################
# 3 - Loading keras model #
###########################
# ic("--- Loading local keras model ---" )
# model = load_model('mnist_model.keras')
######################################
# 3 - Uploading model to HuggingFace #
######################################
ic ("--- Importing HF API ---")
from huggingface_hub import HfApi
ic("--- Settig up HF API ---")
HuggingFace_api = HfApi()
ic("--- Uploading models ---")
HuggingFace_api.upload_folder(
folder_path="saved_models/",
repo_id="manufy/mnist_model_keras",
repo_type="model",
token=token)
########################################################################
# The following are failed attempts to upload the model to HuggingFace #
########################################################################
# upload_folder("manufy/mnist_model_keras")
# not supported in keras 3
# NotImplementedError: Cannot use 'save_pretrained_keras':
# Keras 3.x is not supported.
# Please save models manually and upload them
# using `upload_folder` or `huggingface-cli upload`.
#ic("--- Saving huggingface model ---" )
#huggingface_hub.save_pretrained_keras('saved_model/')
#push_to_hub_keras(model,
# "manufy/mnist_model_keras")
# include_optimizer = True,
# tags = ["object-detection", "some_other_tag"]
#)
|