Spaces:
Runtime error
Runtime error
| import numpy as np | |
| import tensorflow as tf | |
| import gradio as gr | |
| from huggingface_hub import from_pretrained_keras | |
| import cv2 | |
| img_size = 28 | |
| model = from_pretrained_keras("keras-io/keras-reptile") | |
| def read_image(image): | |
| image = tf.convert_to_tensor(image) | |
| image = cv2.resize() | |
| image = image / 127.5 - 1 | |
| return image | |
| def infer(model, image_tensor): | |
| predictions = model.predict(np.expand_dims((image_tensor), axis=0)) | |
| predictions = np.squeeze(predictions) | |
| predictions = np.argmax(predictions, axis=0) | |
| return predictions | |
| def display_result(input_image): | |
| image_tensor = read_image(input_image) | |
| prediction_label = infer(model=model, image_tensor=image_tensor) | |
| return prediction_label | |
| input = gr.inputs.Image() | |
| examples = [["/content/drive/MyDrive/boot.jpg"], ["/content/drive/MyDrive/sneaker.jpg"]] | |
| title = "Few shot learning" | |
| description = "Upload an image or select from examples to classify fashion items." | |
| gr.Interface(display_result, input, outputs="text", examples=examples, allow_flagging=False, analytics_enabled=False, | |
| title=title, description=description).launch(enable_queue=True) |