Update app.py

app.py

@@ -28,12 +28,12 @@ from fudge.util import save_checkpoint, ProgressMeter, AverageMeter, num_params
 from fudge.constants import *
 
 
-device = 'cuda'
+device = 'cpu'
 # imp.reload(model)
 pretrained_model = "checkpoint-150/"
 generation_model = AutoModelForSeq2SeqLM.from_pretrained(pretrained_model, return_dict=True).to(device)
 
-device = 'cuda'
+
 pad_id = 0
 
 generation_model.eval()