Update to PyTorch >= 2.0

app.py

@@ -37,7 +37,7 @@ class App:
     def _get_model(self, name):
         if name in self._model_cache:
             return self._model_cache[name]
-        model = torch.hub.load('baudm/parseq', name, pretrained=True).eval()
+        model = torch.hub.load('baudm/parseq', name, pretrained=True, trust_repo=True).eval()
         self._model_cache[name] = model
         return model
 

requirements.txt

@@ -1,5 +1,8 @@
-Gradio
-torch~=1.13.1
-pytorch-lightning~=1.6.5
-timm~=0.6.7
+--extra-index-url https://download.pytorch.org/whl/cpu
+
+gradio >=3.0.0, <4.0.0
+torch >=2.0.0
+torchvision >=0.15.0
+pytorch-lightning ~=2.2.0
+timm ~=0.9.16
 nltk