ParitoshBorkar commited on
Commit
d975bf6
·
1 Parent(s): 1eafe63

Autogenerate Gradio interface for Toad Frog model prediction

Browse files
Files changed (3) hide show
  1. app.py +5 -3
  2. testing/sample_frog.jpg +0 -0
  3. testing/sample_toad.jpg +0 -0
app.py CHANGED
@@ -3,8 +3,9 @@
3
  # %% auto 0
4
  __all__ = ['toad_frog_model', 'categories', 'image', 'label', 'examples', 'gradio_interface', 'classify_image']
5
 
6
- # %% ../toad_frog_testing.ipynb 2
7
- !pip install -Uqq fastai
 
8
 
9
  # %% ../toad_frog_testing.ipynb 7
10
  toad_frog_model = load_learner("toadfrog.pkl")
@@ -19,7 +20,8 @@ def classify_image(img_path: str):
19
  # %% ../toad_frog_testing.ipynb 14
20
  image = gr.inputs.Image(shape=(192,192))
21
  label = gr.outputs.Label()
22
- examples = ["sample_toad.jpg", "sample_frog.jpg"]
 
23
 
24
  gradio_interface = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
25
  gradio_interface.launch(inline=False)
 
3
  # %% auto 0
4
  __all__ = ['toad_frog_model', 'categories', 'image', 'label', 'examples', 'gradio_interface', 'classify_image']
5
 
6
+ # %% ../toad_frog_testing.ipynb 3
7
+ from fastai.vision.all import *
8
+ import gradio as gr
9
 
10
  # %% ../toad_frog_testing.ipynb 7
11
  toad_frog_model = load_learner("toadfrog.pkl")
 
20
  # %% ../toad_frog_testing.ipynb 14
21
  image = gr.inputs.Image(shape=(192,192))
22
  label = gr.outputs.Label()
23
+ examples = ["testing/sample_toad.jpg", "testing/sample_frog.jpg"]
24
+
25
 
26
  gradio_interface = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
27
  gradio_interface.launch(inline=False)
testing/sample_frog.jpg ADDED
testing/sample_toad.jpg ADDED