Update app.py

app.py

@@ -11,17 +11,17 @@ def is_cat(x): return x[0].isupper()
 
 
 # %% ../drive/MyDrive/Colab Notebooks/cats_vs_dogs.ipynb 12
-learn = load_learner('model.pth')
+learn = load_learner('pets.pkl')
 
 # %% ../drive/MyDrive/Colab Notebooks/cats_vs_dogs.ipynb 14
-categories = ('Bom', 'Ruim')
+categories = ('Dog', 'Cat')
 
 def classify_image(img):
     pred,idx,probs = learn.predict(img)
     return dict(zip(categories, map(float,probs)))
 
 # %% ../drive/MyDrive/Colab Notebooks/cats_vs_dogs.ipynb 15
-image = gr.inputs.Image(shape=(224, 224))
+image = gr.inputs.Image(shape=(192, 192))
 label = gr.outputs.Label()
 examples = ['dog.jpg']