init

app.py

@@ -1,7 +1,6 @@
 import os
 
 import torch
-import torch.nn as nn
 import numpy as np
 import torch.nn.functional as F
 import torchvision.transforms as T
@@ -18,7 +17,7 @@ from transforms import (
 import gradio as gr
 from huggingface_hub import hf_hub_download
 
-class Uniformerv2(nn.Module):
+class Uniformerv2(torch.nn.Module):
     def __init__(self, model):
         super().__init__()
         self.backbone = model