Update app.py

app.py

@@ -67,7 +67,7 @@ config = {
     'batch_size':3,
     'epochs': 1,
     'lr': 1e-5,
-    'device': 'cuda' if torch.cuda.is_available() else 'cpu',
+    'device':  'cpu',
     'scheduler': 'CosineAnnealingWarmRestarts'
 }