Update app.py

app.py

@@ -44,7 +44,7 @@ device = torch.device('cpu')
 parser = argparse.ArgumentParser('PDFNet Testing script', parents=[get_args_parser()])
 args = parser.parse_args(args=[])
 model,model_name = build_model(args)
-model_path = hf_hub_download(repo_id="Tennineee/PDFNet",filename="PDFNet_Best.pth", repo_type="model")
+model_path = hf_hub_download(repo_id="Tennineee/PDFNet-general",filename="PDFNet-Generally.pth", repo_type="model")
 model.load_state_dict(torch.load(model_path,map_location='cpu'),strict=False)
 model = model.to(device).eval()
 
@@ -68,7 +68,7 @@ DAMV2.load_state_dict(state_dict)
 DAMV2 = DAMV2.to(device).eval()
 
 title = "# PDFNet"
-description = """Official demo for **PDFNet**, here use DAMV2-base to generate depth map.
+description = """Official demo for **PDFNet**-general, train on DIS-5K, HRSOD-TR, UHRSD-TR and UHRSD-TE. And here uses DAMV2-base to generate depth map.
 Please refer to our [paper](https://arxiv.org/abs/2503.06100) and [github](https://github.com/Tennine2077/PDFNet) for more details."""
 
 class GOSNormalize(object):