Update app.py

app.py

@@ -31,6 +31,9 @@ if torch.cuda.is_available():
     zh_pretrained_model.half()
     device = torch.device("cuda")
     zh_pretrained_model.to(device)
+else:
+    import torch
+    zh_pretrained_model.to(torch.bfloat16)
 
 
 #en_pretrained_model.eval()