Update app.py

app.py

@@ -34,7 +34,7 @@ def pick_dtype(device: str) -> torch.dtype:
         return torch.bfloat16 if major >= 8 else torch.float16  # Ampere+ -> bf16
     if device == "mps":
         return torch.float16
-    return torch.bfloat16  # CPU
+    return torch.float16  # CPU
 
 def move_to_device(batch, device: str):
     if isinstance(batch, dict):