Update app.py

app.py

@@ -34,6 +34,19 @@ import torch.nn.functional as F
 # learn = load_learner(repo_id)
 #learner = from_pretrained_fastai(repo_id)
 
+
+class ItemTransform(Transform):
+    "A transform that always take tuples as items"
+    _retain = True
+    def __call__(self, x, **kwargs): return self._call1(x, '__call__', **kwargs)
+    def decode(self, x, **kwargs):   return self._call1(x, 'decode', **kwargs)
+    def _call1(self, x, name, **kwargs):
+        if not _is_tuple(x): return getattr(super(), name)(x, **kwargs)
+        y = getattr(super(), name)(list(x), **kwargs)
+        if not self._retain: return y
+        if is_listy(y) and not isinstance(y, tuple): y = tuple(y)
+        return retain_type(y, x)
+
 from huggingface_hub import from_pretrained_fastai
 import torchvision.transforms as transforms
 # from Transform import ItemTransform