yadav commited on
Commit
a7ec989
1 Parent(s): 4bfc96f

updating models

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. traffic_sign_cpu.pt +0 -0
app.py CHANGED
@@ -64,7 +64,7 @@ def transform_images(img):
64
  return transform(img)
65
 
66
 
67
- model = torch.jit.load('model_scripted.pt')
68
  model.eval()
69
 
70
  def classify_image(img):
 
64
  return transform(img)
65
 
66
 
67
+ model = torch.jit.load('traffic_sign_cpu.pt')
68
  model.eval()
69
 
70
  def classify_image(img):
traffic_sign_cpu.pt ADDED
Binary file (396 kB). View file