Update app.py

app.py

@@ -2,11 +2,14 @@ from fastai.vision.all import *
 import gradio as gr
 import glob
 
-class Hook():
-  def hook_func(self, m, i, o): self.stored = o.detach().clone()
+#class Hook():
+#  def hook_func(self, m, i, o): self.stored = o.detach().clone()
 
-learn = load_learner("resnet152_fit_one_cycle_freeze_91acc.pkl", cpu=True)
+#learn = load_learner("resnet152_fit_one_cycle_freeze_91acc.pkl", cpu=True)
 
+#categories = ('arbanasi', 'filibe', 'gjirokoster', 'iskodra', 'kula', 'kuzguncuk', 'larissa_ampelakia', 'mardin', 'ohrid', 'pristina', 'safranbolu', 'selanik', 'sozopol_suzebolu', 'tiran', 'varna')
+
+learn = load_learner("convnext_mixup_0_33.pkl", cpu=True)
 categories = ('arbanasi', 'filibe', 'gjirokoster', 'iskodra', 'kula', 'kuzguncuk', 'larissa_ampelakia', 'mardin', 'ohrid', 'pristina', 'safranbolu', 'selanik', 'sozopol_suzebolu', 'tiran', 'varna')
 def classify_img(img):
   pred,idx,probs=learn.predict(img)