update map_location of model to cpu

app.py

@@ -10,6 +10,8 @@ import matplotlib.pyplot as plt
 import matplotlib.ticker as ticker
 from huggingface_hub import HfApi
 
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+
 def load_checkpoint(filepath):
     """Builds PyTorch Model from saved model
     Returns built model