D-Shepard-p1-57 / example.py
VeduCo's picture
Update example.py
97a07cf verified
import os
os.environ["KERAS_BACKEND"] = "tensorflow"
import keras
from huggingface_hub import hf_hub_download
from PIL import Image
import numpy as np
# 1. Download the model file
print("--- Downloading model... ---")
model_path = hf_hub_download(
repo_id="VeduCo/D-Shepard-p1-57",
filename="ShepardsGift_final_model.keras" # Important: Use the correct filename
)
# 2. Load the model
model = keras.saving.load_model(model_path)
print("--- Model loaded successfully! ---")
model.summary()