Update download_weights.py

Author: alexgenovese

script/download_weights.py

@@ -1,11 +1,12 @@
+#!/usr/bin/env python
 # Run this before you deploy it on replicate, because if you don't
 # whenever you run the model, it will download the weights from the
 # internet, which will take a long time.
 
 import torch
 from diffusers import AutoencoderKL, DiffusionPipeline
 from diffusers.pipelines.stable_diffusion.safety_checker import (
-    StableDiffusionSafetyChecker,
+    StableDiffusionSafetyChecker
 )
 
 # pipe = DiffusionPipeline.from_pretrained(