Spaces:

djl234
/

UFO

Build error

djl234 commited on Jul 21, 2022

Commit

472e711

1 Parent(s): 7b7a2d8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ import collections
 net = build_model('cpu').to('cpu')
     #net=torch.nn.DataParallel(net)
-model_path = 'image_best1.pth'
 print(model_path)
 weight=torch.load(model_path,map_location=torch.device('cpu'))
 #print(type(weight))

 net = build_model('cpu').to('cpu')
     #net=torch.nn.DataParallel(net)
+model_path = 'image_best.pth'
 print(model_path)
 weight=torch.load(model_path,map_location=torch.device('cpu'))
 #print(type(weight))