Hugging Face's logo

Spaces:
sankhyikii
/
A_Dog_Or_A_Cat
Build error

App Files Community
sankhyikii commited on
Commit
bd0fc79
·
1 Parent(s): 89da48e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from timm.data import resolve_data_config
5
  from timm.data.transforms_factory import create_transform
6
 
7
  LABELS = {0:'Cat', 1:'Dog'}
8
- model = torch.load('CatVsDogModel.pth')
9
  model.eval()
10
  transform = create_transform(**resolve_data_config({},model=model))
11
 
 
5
  from timm.data.transforms_factory import create_transform
6
 
7
  LABELS = {0:'Cat', 1:'Dog'}
8
+ model = torch.load('CatVsDogsModel.pth')
9
  model.eval()
10
  transform = create_transform(**resolve_data_config({},model=model))
11