Update app.py

app.py

@@ -7,15 +7,16 @@ from diffusers import DiffusionPipeline
 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
+model_repo_id = "prithivMLmods/SD3.5-Turbo-Realism-2.0-LoRA"
 
-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)
+pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch.float32)
+# pipe = pipe.to(device)
+pipe.to("cpu")
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024