Spaces:

sankhyikii
/

A_Dog_Or_A_Cat

Build error

shyamgupta196 commited on Dec 13, 2022

Commit

325c3c6

•

1 Parent(s): 10367e5

bug fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torch
 from timm.data import resolve_data_config
 from timm.data.transforms_factory import create_transform
-LABELS = [0:'Cat', 1:'Dog']
 model = torch.load('CatVsDogModel.py')
 model.eval()
 transform = create_transform(**resolve_data_config({},model=model))

 from timm.data import resolve_data_config
 from timm.data.transforms_factory import create_transform
+LABELS = {0:'Cat', 1:'Dog'}
 model = torch.load('CatVsDogModel.py')
 model.eval()
 transform = create_transform(**resolve_data_config({},model=model))