{ "cells": [ { "cell_type": "code", "execution_count": 1, "id": "250fcb86", "metadata": {}, "outputs": [], "source": [ "#|default_exp app" ] }, { "cell_type": "markdown", "id": "ea967051", "metadata": {}, "source": [ "# Powerline app" ] }, { "cell_type": "code", "execution_count": 2, "id": "3f54ef95", "metadata": {}, "outputs": [], "source": [ "#|export\n", "from fastai.vision.all import *\n", "import gradio as gr\n", "import skimage" ] }, { "cell_type": "code", "execution_count": 3, "id": "d25179ca", "metadata": {}, "outputs": [], "source": [ "#|export\n", "learn = load_learner('model.pkl')" ] }, { "cell_type": "code", "execution_count": 7, "id": "d752f05b", "metadata": {}, "outputs": [], "source": [ "#|export\n", "categories = ['power line down after storm',\n", " 'tree down on power line']\n", "\n", "def predict(img):\n", " img = PILImage.create(img)\n", " pred,idx,probs = learn.predict(img)\n", " return dict(zip(categories, map(float, probs)))" ] }, { "cell_type": "code", "execution_count": 8, "id": "80d55893", "metadata": { "scrolled": false }, "outputs": [ { "name": "stderr", "output_type": "stream", "text": [ "/Users/hectortarrido-picart/mambaforge/envs/powerline-down-id/lib/python3.8/site-packages/gradio/inputs.py:257: UserWarning: Usage of gradio.inputs is deprecated, and will not be supported in the future, please import your component from gradio.components\n", " warnings.warn(\n", "/Users/hectortarrido-picart/mambaforge/envs/powerline-down-id/lib/python3.8/site-packages/gradio/deprecation.py:40: UserWarning: `optional` parameter is deprecated, and it has no effect\n", " warnings.warn(value)\n", "/Users/hectortarrido-picart/mambaforge/envs/powerline-down-id/lib/python3.8/site-packages/gradio/outputs.py:197: UserWarning: Usage of gradio.outputs is deprecated, and will not be supported in the future, please import your components from gradio.components\n", " warnings.warn(\n", "/Users/hectortarrido-picart/mambaforge/envs/powerline-down-id/lib/python3.8/site-packages/gradio/deprecation.py:40: UserWarning: The 'type' parameter has been deprecated. Use the Number component instead.\n", " warnings.warn(value)\n", "/Users/hectortarrido-picart/mambaforge/envs/powerline-down-id/lib/python3.8/site-packages/gradio/deprecation.py:40: UserWarning: `enable_queue` is deprecated in `Interface()`, please use it within `launch()` instead.\n", " warnings.warn(value)\n" ] }, { "name": "stdout", "output_type": "stream", "text": [ "Running on local URL: http://127.0.0.1:7861\n", "\n", "Could not create share link. Please check your internet connection or our status page: https://status.gradio.app\n" ] }, { "data": { "text/html": [ "
" ], "text/plain": [ "