Edgar404 commited on
Commit
0b2a187
·
verified ·
1 Parent(s): dd44a32

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -40,7 +40,7 @@ models = {'baseline': base_model ,
40
  'performance': performance_model ,
41
  'latence': optimized_model}
42
 
43
- processor = {'baseline': base_processor ,
44
  'performance': performance_processor ,
45
  'latence': optimized_processor}
46
 
@@ -56,7 +56,7 @@ def process_image(image , mode = 'baseline' ):
56
  image : a machine readable image of class PIL or numpy"""
57
 
58
  model = models[mode]
59
- processor = processor[mode]
60
  d_type = torch.float32
61
 
62
  model.to(device)
 
40
  'performance': performance_model ,
41
  'latence': optimized_model}
42
 
43
+ processors = {'baseline': base_processor ,
44
  'performance': performance_processor ,
45
  'latence': optimized_processor}
46
 
 
56
  image : a machine readable image of class PIL or numpy"""
57
 
58
  model = models[mode]
59
+ processor = processors[mode]
60
  d_type = torch.float32
61
 
62
  model.to(device)