Hugging Face's logo

Spaces:
smishr-18
/
FlowerClassification
Sleeping

App Files Community
smishr-18 commited on
Commit
00b9bf6
·
verified ·
1 Parent(s): c85960a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import torchvision.transforms as transforms
5
  from PIL import Image
6
  from resnet import Resnet50Flower102
7
  import pandas as pd
8
- from dataloader import transform
9
  st.title("Flower Image Classification")
10
 
11
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
5
  from PIL import Image
6
  from resnet import Resnet50Flower102
7
  import pandas as pd
8
+
9
  st.title("Flower Image Classification")
10
 
11
  device = 'cuda' if torch.cuda.is_available() else 'cpu'