added cpu shift code again

app.py

@@ -10,6 +10,15 @@ from PIL import Image
 
 
 
+# Moving data to CPU
+def to_device(data, device):
+    """Move tensor(s) to chosen device"""
+    if isinstance(data, (list,tuple)):
+        return [to_device(x, device) for x in data]
+    return data.to(device, non_blocking=True)
+
+
+
 # Defining our Class for just prediction
 def accuracy(outputs, labels):
     _, preds = torch.max(outputs, dim=1)
@@ -60,7 +69,8 @@ classes = ['burger',  'butter_naan',  'chai',  'chapati',  'chole_bhature',
            'kaathi_rolls',  'kadai_paneer',  'kulfi',  'masala_dosa',  'momos',
            'paani_puri',  'pakode',  'pav_bhaji',  'pizza',  'samosa']
 model = IndianFoodModelResnet50(len(classes), pretrained=True)
-
+device = 'cpu'
+to_device(model, device);
 
 
 # Loading the model