Rick458 commited on
Commit
fd7be80
1 Parent(s): 729c782

updated gpu/cpu device code

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -98,6 +98,7 @@ classes = ['burger', 'butter_naan', 'chai', 'chapati', 'chole_bhature',
98
  'kaathi_rolls', 'kadai_paneer', 'kulfi', 'masala_dosa', 'momos',
99
  'paani_puri', 'pakode', 'pav_bhaji', 'pizza', 'samosa']
100
  model = IndianFoodModelResnet50(len(classes), pretrained=True)
 
101
  to_device(model, device);
102
 
103
 
 
98
  'kaathi_rolls', 'kadai_paneer', 'kulfi', 'masala_dosa', 'momos',
99
  'paani_puri', 'pakode', 'pav_bhaji', 'pizza', 'samosa']
100
  model = IndianFoodModelResnet50(len(classes), pretrained=True)
101
+ device = get_default_device()
102
  to_device(model, device);
103
 
104