Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from fastai.vision.all import load_learner, PILImage | |
| import os | |
| import pathlib | |
| if os.name == "nt": # Windows | |
| pathlib.PosixPath = pathlib.WindowsPath | |
| learn = load_learner("cards_classification.pkl", cpu=True) | |
| def classify_image(image): | |
| if image is None: | |
| return {} | |
| img = PILImage.create(image) | |
| pred_class, pred_idx, probs = learn.predict(img) | |
| return {learn.dls.vocab[i]: float(probs[i]) for i in range(len(probs))} | |
| # Create the interface | |
| with gr.Blocks() as iface: | |
| gr.Markdown( | |
| """ | |
| # 🃏 Playing Card Image Classifier | |
| This AI model can recognize and classify images of standard playing cards using deep learning. | |
| ### How it works: | |
| Upload an image of a playing card, and the model will predict which card it is from a full deck of cards. | |
| ### Supported Cards: | |
| The model can identify **all standard playing cards**, including: | |
| - Aces, numbers (2–10), Jacks, Queens, and Kings | |
| - All four suits: Clubs, Diamonds, Hearts, and Spades | |
| - Joker card (In total: 53 classes) | |
| ### Tips for best results: | |
| - Make sure the card is fully visible | |
| - Avoid heavy glare, blur, or occlusion | |
| - One card per image works best | |
| """ | |
| ) | |
| with gr.Row(): | |
| with gr.Column(): | |
| input_image = gr.Image(type="pil", label="Upload Card Image") | |
| submit_btn = gr.Button("🔍 Identify Card", variant="primary") | |
| with gr.Column(): | |
| output_label = gr.Label(num_top_classes=3, label="Predictions") | |
| gr.Examples( | |
| examples=[ | |
| ["test_images/king_of_diamonds.jpg"], | |
| ["test_images/jack_of_spades.jpg"], | |
| ["test_images/ace_of_diamonds.jpg"], | |
| ["test_images/three_of_hearts.jpg"], | |
| ["test_images/ace_of_spades.jpg"] | |
| ], | |
| inputs=input_image, | |
| label="Try an example" | |
| ) | |
| submit_btn.click(fn=classify_image, inputs=input_image, outputs=output_label) | |
| if __name__ == "__main__": | |
| iface.launch(server_name="0.0.0.0", server_port=7860) |