islasher commited on
Commit
aa505fc
1 Parent(s): ce67da6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -3,8 +3,8 @@
3
  # from huggingface_hub import from_pretrained_fastai
4
  import gradio as gr
5
 
 
6
  from fastai.data.block import DataBlock
7
-
8
  from fastai.data.transforms import get_image_files, FuncSplitter, Normalize
9
  from fastai.layers import Mish
10
  from fastai.losses import BaseLoss
@@ -36,7 +36,8 @@ import torch.nn.functional as F
36
 
37
  from huggingface_hub import from_pretrained_fastai
38
  import torchvision.transforms as transforms
39
- from Transform import ItemTransform
 
40
 
41
  from albumentations import (
42
  Compose,
 
3
  # from huggingface_hub import from_pretrained_fastai
4
  import gradio as gr
5
 
6
+ from fastai import *
7
  from fastai.data.block import DataBlock
 
8
  from fastai.data.transforms import get_image_files, FuncSplitter, Normalize
9
  from fastai.layers import Mish
10
  from fastai.losses import BaseLoss
 
36
 
37
  from huggingface_hub import from_pretrained_fastai
38
  import torchvision.transforms as transforms
39
+ # from Transform import ItemTransform
40
+
41
 
42
  from albumentations import (
43
  Compose,