Update app.py

app.py

@@ -12,8 +12,11 @@ from rudolph import utils
 from PIL import Image
 
 device = 'cpu'
-
-model = get_rudolph_model('350M',  fp16=True, device=device)
+if device=='cuda':
+  half = True
+else:
+  half = False
+model = get_rudolph_model('350M',  fp16=half, device=device)
 model.load_state_dict(torch.load("awesomemodel__dalle_1500.pt",map_location=torch.device('cpu')))
 tokenizer = get_tokenizer()
 vae = get_vae(dwt=False).to(device)