gradio web app ready

app.py

@@ -1,27 +1,32 @@
-# AUTOGENERATED! DO NOT EDIT! File to edit: . (unless otherwise specified).
 
-__all__ = ['is_cat', 'learn', 'classify_image', 'categories', 'image', 'label', 'examples', 'intf']
-
-# Cell
+# %%
 from fastai.vision.all import *
 import gradio as gr
 
 def is_cat(x): return x[0].isupper()
 
-# Cell
-learn = load_learner('model.pkl')
 
-# Cell
-categories = ('Dog', 'Cat')
+
+
+
+
+# %%
+learn = load_learner('export.pkl')
+
+# %%
+categories = ('desktop pc','laptop','mobile phone')
 
 def classify_image(img):
     pred,idx,probs = learn.predict(img)
     return dict(zip(categories, map(float,probs)))
 
-# Cell
+# %%
 image = gr.inputs.Image(shape=(192, 192))
 label = gr.outputs.Label()
-examples = ['dog.jpg', 'cat.jpg', 'dunno.jpg']
+examples = ['dell 20.jpg', 'dell inspiron.jpg', 'desktop pc on table.jpg']
 
 intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
-intf.launch(inline=False)
+intf.launch(inline=False)
+
+# %%
+## end -

converter_update.py

@@ -15,7 +15,7 @@ py_file = open(f"{filename}.py", "w+")
 
 for cell in code['cells']:
     if cell['cell_type'] == 'code':
-        if cell['source'] and "#| export" in cell['source'][0]:
+        if cell['source'] and (("#| export" in cell['source'][0]) or("#|export" in cell['source'][0]) ):
             py_file.write("# %%\n")       #you can comment out this line
             for line in cell['source'][1:]:
                 py_file.write(line)