gpu to cpu_2

app.py

@@ -67,7 +67,7 @@ transformer = Transformer(d_model,
                           PADDING_TOKEN)
 
 model_file = hf_hub_download(repo_id="yashAI007/English_to_Gujarati_Translation", filename="model.pth")
-model = torch.load(model_file)
+model = torch.load(model_file,map_location='cpu')
 transformer.load_state_dict(model['model_state_dict'])
 transformer.to(device)
 transformer.eval()