Hugging Face's logo

Spaces:
sankhyikii
/
A_Dog_Or_A_Cat
Build error

App Files Community
shyamgupta196 commited on
Commit
956b27f
•
1 Parent(s): c27b177

eval rem for pretrained

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