Spaces:
Build error
Build error
| import gradio as gr | |
| from inference import load_model, run_inference | |
| net_feat, net_cls = load_model() | |
| def classify_image(image): | |
| return run_inference(image, net_feat, net_cls) | |
| example_images = ["example.jpeg", "example2.webp","image2.jpg"] | |
| interface = gr.Interface( | |
| fn=classify_image, | |
| inputs=gr.Image(type="filepath"), | |
| outputs=gr.Textbox(label="Predicted clothing type:"), | |
| title="Clothing1M Classifier", | |
| description="Upload an image of clothing image.", | |
| examples=example_images | |
| ) | |
| if __name__ == "__main__": | |
| interface.launch() |