Update inference.py

inference.py

@@ -14,6 +14,10 @@ from model import NetG, CLIP_TXT_ENCODER
 
 # checking the device
 device = 'cuda:0' if torch.cuda.is_available() else 'cpu'
+
+# Getting the HF token
+HF_TOKEN = os.getenv("HF_TOKEN")
+
 # repositiory of the model
 repo_id = "VinayHajare/EfficientCLIP-GAN"
 file_name = "saved_models/state_epoch_120.pth"