tweak to file structure

simple_image_classifier/app.py → app.py

@@ -1,10 +1,10 @@
-# AUTOGENERATED! DO NOT EDIT! File to edit: ../four-way-classifier.ipynb.
+# AUTOGENERATED! DO NOT EDIT! File to edit: four-way-classifier.ipynb.
 
 # %% auto 0
 __all__ = ['learn', 'categories', 'model', 'description', 'image', 'label', 'examples', 'interf', 'get_best_layout',
            'create_plots', 'classify_image']
 
-# %% ../four-way-classifier.ipynb 1
+# %% four-way-classifier.ipynb 1
 from fastai.vision.all import *
 import gradio as gr
 import PIL
@@ -15,14 +15,14 @@ import primefac
 from torchviz import make_dot
 from pytorch_grad_cam.utils.image import preprocess_image
 
-# %% ../four-way-classifier.ipynb 2
+# %% four-way-classifier.ipynb 2
 warnings.filterwarnings( 'ignore' )
 warnings.simplefilter( 'ignore' )
 
 # plt.rcParams[ 'figure.figsize' ] = [ 15, 15 ]
 plt.rcParams[ 'figure.autolayout' ] = True
 
-# %% ../four-way-classifier.ipynb 3
+# %% four-way-classifier.ipynb 3
 def get_best_layout( num_imgs, img_size ):
     prime_factors = list( primefac.primefac( num_imgs ) )
     aspect_ratios = []
@@ -73,12 +73,12 @@ def classify_image( img ):
     
     return ( dict( zip( categories, map( float, probs ) ) ), im1, im2, im3 )
 
-# %% ../four-way-classifier.ipynb 4
+# %% four-way-classifier.ipynb 4
 learn = load_learner( 'simple-image-classifier.pkl' )
 categories = [ 'bird', 'forest', 'otter', 'snake' ]
 model = learn.model.eval()
 
-# %% ../four-way-classifier.ipynb 5
+# %% four-way-classifier.ipynb 5
 description='''
 A simple 4-way classifier that categorizes images as 'snake', 'bird', 'otter' or 'forest'. 
 Refined from an initial ResNet18 model downloaded from HuggingFace.

simple_image_classifier/.ipynb_checkpoints/app-checkpoint.py

@@ -1,15 +0,0 @@
-# AUTOGENERATED! DO NOT EDIT! File to edit: ../gradio_test.ipynb.
-
-# %% auto 0
-__all__ = ['iface', 'greet']
-
-# %% ../gradio_test.ipynb 1
-import gradio as gr 
-
-# %% ../gradio_test.ipynb 3
-def greet( name ): 
-    return 'Hello %s!!'%name
-    
-iface = gr.Interface( fn=greet, inputs='text', outputs='text' ) 
-iface.launch( inline=True )
-