import numpy as np

import gradio as gr
from models import efficientnetv2b0_25_arch_styles_Classifier


demo = gr.Interface(
<<<<<<< HEAD
    efficientnetv2b0_25_arch_styles_Classifier, "image", "label")
=======
    efficientnetv2b0_25_arch_styles_Classifier, "image", outputs=["text"])
>>>>>>> 6dfb3eb853ff6baf202a3117ebb5b61755454768

demo.launch()