Update app.py

app.py

@@ -29,18 +29,16 @@ def load_pipelines():
     pipe = pipe.to(device)
     PIPELINES[mid] = pipe
 
-# load_pipelines()
 
-# 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)
+load_pipelines()
 
-# MAX_SEED = np.iinfo(np.int32).max
-# MAX_IMAGE_SIZE = 1024
+MAX_SEED = np.iinfo(np.int32).max
+MAX_IMAGE_SIZE = 1024
 
 
 @spaces.GPU #[uncomment to use ZeroGPU]