| import gradio as gr | |
| from transformers import pipeline | |
| pipeline = pipeline(task="image-classification", model="pranavanbupathy/test-cifar-10") | |
| def predict(image): | |
| predictions = pipeline(image) | |
| result = predictions[0] | |
| return result["label"] | |
| gr.Interface( | |
| predict, | |
| examples=["./Bombardier-CRJ-700.jpg","./atr72.jpeg","./c208.jpg","./fokker70.jpg","b737.jpg"], | |
| inputs=gr.inputs.Image(label="Upload only commercial aircraft image here", type="filepath"), | |
| outputs=gr.outputs.Label(), | |
| title="Find which Aircraft it is ✈️", | |
| ).launch(inline=False) | |