Update app.py

app.py

@@ -3,11 +3,11 @@ import gradio as gr
 import numpy as np
 from PIL import Image
 import torchvision.transforms as T
-from model import HybridDepthModel
+from model import DepthSTAR
 
 # Load model
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = HybridDepthModel().to(device)
+model = DepthSTAR().to(device)
 model.load_state_dict(torch.load("depth_model_all.pth", map_location=device))
 model.eval()