fix: bug

index.py

@@ -4,7 +4,7 @@ import gradio as gr
 import model
 import torch
 
-gr.info(f"⏳ Downloading model from huggingface hub...")
+gr.Info(f"⏳ Downloading model from huggingface hub...")
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model = model.BEN_Base().to(device).eval() #init pipeline
 
@@ -14,15 +14,15 @@ model_path = hf_hub_download(
     filename="BEN_Base.pth",
     cache_dir="./models"  # 缓存目录，避免重复下载
 )
-gr.info(f"✅ Model downloaded successfully to {model_path}")
+gr.Info(f"✅ Model downloaded successfully to {model_path}")
 model.loadcheckpoints(model_path)
 
 
 
 def handler(input_image):  
-    gr.info("🚀 Processing image...")
+    gr.Info("🚀 Processing image...")
     mask, foreground = model.inference(input_image)
-    gr.info("✅ Image processing completed!")
+    gr.Info("✅ Image processing completed!")
 
     return [mask, foreground]