Update app.py

app.py

@@ -19,6 +19,9 @@ import transformers
 transformers.utils.move_cache()
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
+if "HF_TOKEN_LOGIN" in os.environ:
+    login(token=os.environ["HF_TOKEN_LOGIN"])
+
 def infer(path_input, seed):
     name_base, name_ext = os.path.splitext(os.path.basename(path_input))
     output_g, output_d = lotus(path_input, 'depth', seed, device)