Update app.py

app.py

@@ -22,7 +22,7 @@ map_location = torch.device("cuda") if enable_gpu else "cpu"
 from huggingface_hub import hf_hub_download
 from fastai.learner import load_learner
 
-model = torch.load("h2z-85epoch.pth")
+model = torch.load("h2z-85epoch.pth",map_location=torch.device('cpu'))
 def generate_img(img_path):
     img = tf.io.read_file(img_path)
     img = tf.image.decode_png(img)