init

app.py

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

requirements.txt

@@ -1,4 +1,4 @@
-torch
+torch==1.9.1
 torchvision 
 einops 
 timm