Spaces:
Runtime error
Runtime error
updated gpu/cpu device code
Browse files
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 |
|