Update app.py

app.py

@@ -27,9 +27,8 @@ DEVICE = "cuda" if torch.cuda.is_available() else "CPU"
 # ).to(DEVICE)
 
 
-
-from .linear_head import LinearPts3d
-from .dpt_head import create_dpt_head
+from mini_dust3r.heads.linear_head import LinearPts3d
+from mini_dust3r.heads.dpt_head import create_dpt_head
 
 def head_factory(head_type, output_mode, net, has_conf=False):
     """" build a prediction head for the decoder