Update app.py

app.py

@@ -9,10 +9,10 @@ import torch
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_repo_id = "stabilityai/sdxl-turbo"  # Replace to the model you would like to use
 
-#if torch.cuda.is_available():
-torch_dtype = torch.float16
-#else:
-    #torch_dtype = torch.float32
+if torch.cuda.is_available():
+    torch_dtype = torch.float16
+else:
+    torch_dtype = torch.float32
 
 pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
 pipe = pipe.to(device)