import numpy as np import gradio as gr from huggingface_hub import from_pretrained_fastai learn = from_pretrained_fastai('mindwrapped/pokemon-card-checker') def check_card(input_img): pred = learn.predict(input_img) return pred demo = gr.Interface( fn=check_card, inputs=gr.Image(shape=(256, 256)), outputs="text", examples=['real-1.jpeg','real-2.jpeg','fake-1.jpeg','fake-2.jpeg'], title='Pokemon Card Checker', description='A resnet34 model fine tuned to determine whether Pokemon cards are real or fake. Dataset is located [here](https://www.kaggle.com/datasets/ongshujian/real-and-fake-pokemon-cards).') demo.launch()