Use CPU if no GPU

app.py

@@ -32,9 +32,10 @@ config = Config(
   use_w=True,
   batch_size=5_000, # style layer quite small
 )
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 inst = get_instrumented_model(config.model, config.output_class,
-                              config.layer, torch.device('cuda'), use_w=config.use_w)
+                              config.layer, torch.device(device), use_w=config.use_w)
 
 path_to_components = get_or_compute(config, inst)
 

requirements.txt

@@ -8,5 +8,5 @@ torchvision==0.10.0
 torchtext==0.10.0
 scikit_learn==1.0
 scikit-image==0.18.3
-tqdm==4.49.0
+tqdm
 gdown