Update app.py

app.py

@@ -3,8 +3,8 @@
 # from huggingface_hub import from_pretrained_fastai
 import gradio as gr
 
+from fastai import *
 from fastai.data.block import DataBlock
-
 from fastai.data.transforms import get_image_files, FuncSplitter, Normalize
 from fastai.layers import Mish
 from fastai.losses import BaseLoss
@@ -36,7 +36,8 @@ import torch.nn.functional as F
 
 from huggingface_hub import from_pretrained_fastai
 import torchvision.transforms as transforms
-from Transform import ItemTransform
+# from Transform import ItemTransform
+
 
 from albumentations import (
     Compose,