将githun中的内容迁移到huggingface

app.py

@@ -6,13 +6,14 @@
 # @Software: PyCharm
 
 from glob import glob
+
+import albumentations as albu
 import gradio as gr
-import torch
 import numpy as np
-import cv2
+import torch
 from PIL import Image
-import albumentations as albu
 from albumentations.pytorch.transforms import ToTensorV2
+
 from src.models.components.cdnetv1 import CDnetV1
 from src.models.components.cdnetv2 import CDnetV2
 from src.models.components.dual_branch import Dual_Branch
@@ -30,7 +31,7 @@ class Application:
             "hrcloud": HRcloudNet(num_classes=2).to(self.device),
             "mcdnet": MCDNet(in_channels=3, num_classes=2).to(self.device),
             "scnn": SCNNNet(num_classes=2).to(self.device),
-            "dual_branch": Dual_Branch(img_size=256, in_channels=3, num_classes=2).to(
+            "dbnet": Dual_Branch(img_size=256, in_channels=3, num_classes=2).to(
                 self.device
             ),
         }
@@ -116,14 +117,14 @@ class Application:
             [
                 gr.Image(sources=["clipboard", "upload"], type="pil"),
                 gr.Radio(
-                    ["cdnetv1", "cdnetv2", "hrcloud", "mcdnet", "scnn", "dual_branch"],
+                    ["cdnetv1", "cdnetv2", "hrcloud", "mcdnet", "scnn", "dbnet"],
                     label="model_name",
                     info="选择使用的模型",
                 ),
             ],
             [gr.Image(), gr.Textbox(label="提示信息")],
             examples=[
-                ["examples_png/barren_11.png", "dual_branch"],
+                ["examples_png/barren_11.png", "dbnet"],
                 ["examples_png/snow_10.png", "scnn"],
                 ["examples_png/vegetation_21.png", "cdnetv2"],
                 ["examples_png/water_22.png", "hrcloud"],