rahuketu86 commited on
Commit
64d3902
·
verified ·
1 Parent(s): 93e8667

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. app.py +2 -1
  2. requirements.txt +4 -2
app.py CHANGED
@@ -8,6 +8,7 @@ import os
8
  import gradio as gr
9
  from fastai.vision.all import *
10
  import pathlib
 
11
  # import dill
12
 
13
  # %% app.ipynb 5
@@ -20,7 +21,7 @@ dblock = DataBlock(
20
  blocks=(ImageBlock, CategoryBlock(vocab=vocab)),
21
  get_items=lambda x: [img_path], # x is source, ignored here
22
  get_y=lambda x: 'normal',
23
- item_tfms=Resize(192)
24
  )
25
 
26
  # Pass a dummy 'source' argument, e.g. '.' or pathlib.Path('.')
 
8
  import gradio as gr
9
  from fastai.vision.all import *
10
  import pathlib
11
+ import timm
12
  # import dill
13
 
14
  # %% app.ipynb 5
 
21
  blocks=(ImageBlock, CategoryBlock(vocab=vocab)),
22
  get_items=lambda x: [img_path], # x is source, ignored here
23
  get_y=lambda x: 'normal',
24
+ item_tfms=[Resize(192, method='squish')]
25
  )
26
 
27
  # Pass a dummy 'source' argument, e.g. '.' or pathlib.Path('.')
requirements.txt CHANGED
@@ -2,5 +2,7 @@
2
  pillow
3
  fastai
4
  nbdev
5
- huggingface_hub[fastai]
6
- dill
 
 
 
2
  pillow
3
  fastai
4
  nbdev
5
+ huggingface_hub
6
+ fastai
7
+ dill
8
+ timm