Spaces:
Runtime error
Runtime error
File size: 492 Bytes
525f3dd e3cece8 40fa99d cffbee3 e3cece8 5b0a672 525f3dd ed88d55 7854bc4 ed88d55 525f3dd ed88d55 de95e10 525f3dd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
import gradio as gr
import tensorflow as tf
import numpy as np
from huggingface_hub import from_pretrained_keras
def greet(image):
model = from_pretrained_keras("vahidrezanezhad/sbb_binarization")
prediction = model.predict(image.reshape(1,224,448,image.shape[2]))
prediction = tf.squeeze(tf.round(prediction))
print(prediction.shape)
#print(model.summary())
return "Hello "
iface = gr.Interface(fn=greet, inputs=gr.Image(), outputs="text")
iface.launch()
|