Update app.py

app.py

@@ -26,7 +26,7 @@ device = "cuda"
 _dtype = torch.bfloat16
 hf_token = os.getenv("HF_TOKEN")
 
-# init ovis_image
+print("init ovis_image")
 model_config = ovis_image_configs["ovis-image-7b"]
 ovis_image = OvisImageModel(model_config)
 ovis_image_path = hf_hub_download(
@@ -41,7 +41,7 @@ print(f"Load Unexpected Keys {unexpected_keys}")
 ovis_image = ovis_image.to(device=device, dtype=_dtype)
 ovis_image.eval()
 
-# init vae
+print("init vae")
 vae_path = hf_hub_download(
     repo_id="AIDC-AI/Ovis-Image-7B", 
     filename="ae.safetensors", 
@@ -56,10 +56,10 @@ autoencoder = load_ae(
 )
 autoencoder.eval()
 
-# init ovis
+print("init ovis")
 ovis_path = hf_hub_download(
     repo_id="AIDC-AI/Ovis-Image-7B", 
-    filename="Ovis2.5-2B", 
+    subfolder="Ovis2.5-2B", 
     token=hf_token,
 )
 ovis_tokenizer = build_ovis_tokenizer(ovis_path)