Create app.py

app.py

@@ -0,0 +1,17 @@
+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')