Update main.py

main.py

@@ -21,7 +21,7 @@ model_id = "CompVis/stable-diffusion-v1-4"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 pipe = StableDiffusionPipeline.from_pretrained(
-    model_id, torch_dtype=torch.float32, cache_dir="/cache/"
+    model_id, torch_dtype=torch.float32, cache_dir="./cache"
 )
 
 # MOdel 1.0