diff --git a/hugging_face/README.md b/hugging_face/README.md deleted file mode 100644 index b059204..0000000 --- a/hugging_face/README.md +++ /dev/null @@ -1,13 +0,0 @@ ---- -title: Cloud Adapter -emoji: 🏢 -colorFrom: red -colorTo: purple -sdk: gradio -sdk_version: 5.6.0 -app_file: app.py -pinned: false -license: apache-2.0 ---- - -Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference diff --git a/hugging_face/app.py b/hugging_face/app.py index e6c748e..3523170 100644 --- a/hugging_face/app.py +++ b/hugging_face/app.py @@ -22,17 +22,6 @@ def __init__(self, config_path=None, checkpoint_path=None, device="cpu", example self.img_size = 256 if num_classes == 2 else 512 self.palette = palette self.legend = self.html_legend(num_classes=num_classes) - self.name_mapping = { - "KappaMask": "kappamask", - "CDNetv1": "cdnetv1", - "CDNetv2": "cdnetv2", - "HRCloudNet": "hrcloudnet", - "MCDNet": "mcdnet", - "SCNN": "scnn", - "DBNet": "dbnet", - "UNetMobv2": "unetmobv2", - "Cloud-Adapter": "cloud-adapter", - } self.create_ui() diff --git a/hugging_face/requirements.txt b/hugging_face/requirements.txt deleted file mode 100644 index 477b3a5..0000000 --- a/hugging_face/requirements.txt +++ /dev/null @@ -1,24 +0,0 @@ -einops==0.8.0 -segmentation_models_pytorch -gradio==4.44.1 -huggingface-hub==0.26.0 -image_dehazer==0.0.9 - -mmcv==2.1.0 --f https://download.openmmlab.com/mmcv/dist/cpu/torch2.0/index.html - -mmdet==3.3.0 -mmengine==0.10.5 -mmpretrain==1.2.0 -mmsegmentation==1.2.2 -timm==0.9.2 - - -torch==2.0.1 --f https://download.pytorch.org/whl/cpu -torchvision==0.15.2 -torchaudio==2.0.2 -ftfy -regex -yacs -numpy==1.24.2