Spaces:
Runtime error
Runtime error
import gradio as gr | |
import numpy as np | |
from huggingface_hub import from_pretrained_keras | |
siamese = from_pretrained_keras("rushic24/keras-siamese-contrastive") | |
def predict_image(img1, img2): | |
assert img1.shape == (28, 28) | |
assert img1.shape == img2.shape | |
print('img 1 shape', img1.shape) | |
img1 = np.expand_dims(img1, 0) | |
img2 = np.expand_dims(img2, 0) | |
lab = str(siamese.predict([img1, img2])[0][0]) | |
return lab | |
iface = gr.Interface(predict_image, inputs=["sketchpad", "sketchpad"], outputs="label") | |
iface.launch(debug='True') | |