File size: 499 Bytes
97a07cf |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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()
|