{
"cells": [
{
"cell_type": "code",
"execution_count": null,
"id": "c483e658-eca5-48e4-ae77-d59403453d12",
"metadata": {},
"outputs": [],
"source": [
"#|default_exp app"
]
},
{
"cell_type": "code",
"execution_count": 2,
"id": "9ffc7d71-f919-4a81-9ba0-46edbe0c1f99",
"metadata": {},
"outputs": [],
"source": [
"#|export\n",
"\n",
"from fastai.vision.all import *\n",
"import gradio as gr"
]
},
{
"cell_type": "code",
"execution_count": 3,
"id": "06a2e43c-5935-45c7-aee1-bbe8594aff01",
"metadata": {},
"outputs": [],
"source": [
"#|export\n",
"learn = load_learner('model.pkl')"
]
},
{
"cell_type": "code",
"execution_count": 4,
"id": "b43a8038-20d4-41c1-b9c9-c05b35f2a491",
"metadata": {},
"outputs": [],
"source": [
"#|export\n",
"categories = ('Cloudy','Rain','Shine Or Partly Cloudy','Sunrise') \n",
"def classify_image(img):\n",
" pred,idx,probs = learn.predict(img)\n",
" return dict(zip(categories,map(float,probs)))"
]
},
{
"cell_type": "code",
"execution_count": 6,
"id": "49d4264f-ae56-4ac0-b0a4-a876385f5ee2",
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"\n",
"\n"
],
"text/plain": [
""
]
},
"metadata": {},
"output_type": "display_data"
},
{
"data": {
"text/html": [],
"text/plain": [
""
]
},
"metadata": {},
"output_type": "display_data"
},
{
"data": {
"text/plain": [
"('Rain', tensor(1), tensor([7.6823e-04, 9.9922e-01, 1.0592e-05, 4.4878e-08]))"
]
},
"execution_count": 6,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"learn.predict('rain.jpg')"
]
},
{
"cell_type": "code",
"execution_count": 7,
"id": "a8794806-a335-4331-978a-2d8ba3ba8bff",
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Running on local URL: http://127.0.0.1:7860\n",
"\n",
"To create a public link, set `share=True` in `launch()`.\n"
]
},
{
"data": {
"text/plain": []
},
"execution_count": 7,
"metadata": {},
"output_type": "execute_result"
},
{
"data": {
"text/html": [
"\n",
"\n"
],
"text/plain": [
""
]
},
"metadata": {},
"output_type": "display_data"
},
{
"data": {
"text/html": [],
"text/plain": [
""
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"#|export\n",
"#image = gr.inputs.Image(shape=(192,192))\n",
"#label = gr.outputs.Label()\n",
"examples = ['cloudy.jpg','rain.jpg','shine.jpg','sunrise.jpg']\n",
"\n",
"intf = gr.Interface(fn=classify_image , inputs='image' , outputs = 'label' , examples=examples)\n",
"intf.launch(inline=False)"
]
},
{
"cell_type": "code",
"execution_count": 9,
"id": "6f6d0837-0f14-4d58-8ce3-13b7e25376b4",
"metadata": {},
"outputs": [
{
"ename": "ImportError",
"evalue": "cannot import name 'nbdev_export' from 'nbdev.export' (/home/shoaib/mambaforge/lib/python3.10/site-packages/nbdev/export.py)",
"output_type": "error",
"traceback": [
"\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
"\u001b[0;31mImportError\u001b[0m Traceback (most recent call last)",
"Cell \u001b[0;32mIn[9], line 1\u001b[0m\n\u001b[0;32m----> 1\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mnbdev\u001b[39;00m\u001b[38;5;21;01m.\u001b[39;00m\u001b[38;5;21;01mexport\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m nbdev_export\n",
"\u001b[0;31mImportError\u001b[0m: cannot import name 'nbdev_export' from 'nbdev.export' (/home/shoaib/mambaforge/lib/python3.10/site-packages/nbdev/export.py)"
]
}
],
"source": [
"from nbdev.export import notebook2script"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "9fa6c39c-a5e5-42e8-a4c3-122d6bdbeef1",
"metadata": {},
"outputs": [],
"source": []
}
],
"metadata": {
"kernelspec": {
"display_name": "Python 3 (ipykernel)",
"language": "python",
"name": "python3"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.10.12"
}
},
"nbformat": 4,
"nbformat_minor": 5
}