Update gradio_app.py

gradio_app.py

@@ -32,19 +32,19 @@ def single_inference(user_prompt, save_path, guidance_scale, num_sampling_steps,
 
     # Load models
     vae = AllegroAutoencoderKL3D.from_pretrained(
-        "weights_dir/vae/", 
+        "./allegro_weights/vae/", 
         torch_dtype=torch.float32
     ).cuda()
     vae.eval()
 
-    text_encoder = T5EncoderModel.from_pretrained("weights_dir/text_encoder/", torch_dtype=dtype)
+    text_encoder = T5EncoderModel.from_pretrained("./allegro_weights/text_encoder/", torch_dtype=dtype)
     text_encoder.eval()
 
-    tokenizer = T5Tokenizer.from_pretrained("weights_dir/tokenizer/")
+    tokenizer = T5Tokenizer.from_pretrained("./allegro_weights/tokenizer/")
 
     scheduler = EulerAncestralDiscreteScheduler()
 
-    transformer = AllegroTransformer3DModel.from_pretrained("weights_dir/transformer/", torch_dtype=dtype).cuda()
+    transformer = AllegroTransformer3DModel.from_pretrained("./allegro_weights/transformer/", torch_dtype=dtype).cuda()
     transformer.eval()
 
     allegro_pipeline = AllegroPipeline(