{ "cells": [ { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "#|default_exp app" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "%pip install gradio==3.50 fastai ipywidgets" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "#|export\n", "import sys\n", "import subprocess\n", "\n", "subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'gradio==3.50'])\n", "subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'fastai'])\n", "subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'ipywidgets'])\n", "\n", "from fastai.vision.all import *\n", "import gradio as gr\n", "import pathlib as pl\n", "plt = platform.system()\n", "\n", "def is_cat(x): return x[0].isupper()" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "im = PILImage.create('dog.jpg')\n", "im.thumbnail((192,192))\n", "im" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "#|export\n", "if plt == 'Linux' : pl.WindowsPath = pl.PosixPath\n", "learn = load_learner('model.pkl')" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "learn.predict(im)" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "#|export\n", "categories = ('Dog', 'Cat')\n", "\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": null, "metadata": {}, "outputs": [], "source": [ "classify_image(im)" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "#|export\n", "image = gr.Image(height=192, width = 192)\n", "label = gr.Label()\n", "examples = ['dog.jpg', 'cat.jpg', 'dunno.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": 28, "metadata": {}, "outputs": [], "source": [ "'''\n", "import jupytext\n", "nb = jupytext.read(\"app.ipynb\")\n", "jupytext.write(nb, \"app.py\", fmt = \"qmd\")\n", "'''\n", "from nbdev.export import nb_export\n", "\n", "nb_export('app.ipynb')\n" ] } ], "metadata": { "kernelspec": { "display_name": "tutorial-env", "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": 2 }