Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from fastai.vision.all import load_learner, PILImage | |
| # Load your exported fastai model | |
| model_path = "export.pkl" # your exported .pkl | |
| learn = load_learner(model_path) | |
| # Prediction function | |
| def predict(image): | |
| img = PILImage.create(image) # ensure it's a fastai PILImage | |
| pred_class, pred_idx, outputs = learn.predict(img) | |
| return (str(pred_class)) | |
| # Gradio interface | |
| iface = gr.Interface( | |
| fn=predict, | |
| inputs=gr.Image(type="pil"), | |
| outputs=gr.Textbox(label="Prediction"), | |
| title="FastAI Image Classification", | |
| description="Upload an image to get prediction from your exported fastai model." | |
| ) | |
| if __name__ == "__main__": | |
| iface.launch(server_name="0.0.0.0", server_port=7860) | |