Update gradio_app.py

gradio_app.py

@@ -8,12 +8,12 @@ from allegro.pipelines.pipeline_allegro import AllegroPipeline
 from allegro.models.vae.vae_allegro import AllegroAutoencoderKL3D
 from allegro.models.transformers.transformer_3d_allegro import AllegroTransformer3DModel
 
-import huggingface_hub
+from huggingface_hub import snapshot_download
 
 weights_dir = './allegro_weights'
 os.makedirs(weights_dir, exist_ok=True)
 
-huggingface_hub.snapshot_download(
+snapshot_download(
     repo_id='rhymes-ai/Allegro',
     allow_patterns=[
         'scheduler/**',