Update app.py

app.py

@@ -6,6 +6,14 @@ import random
 import logging
 import numpy as np
 from typing import Any, Dict, List, Optional, Union
+from huggingface_hub import login
+
+hf_token = os.getenv("myKey")
+
+if hf_token:
+    login(token=hf_token)
+else:
+    raise ValueError("Hugging Face API token not found. Set HUGGINGFACEHUB_API_TOKEN as a secret.")
 
 import torch
 from PIL import Image