Moving the auth token to where the model is loaded

app.py

@@ -8,6 +8,8 @@ import time
 import gradio as gr
 from transformers import AutoModelForCausalLM
 
+hf_token = os.getenv('HF_TOKEN')
+
 device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
 
 parser = argparse.ArgumentParser(description='Generative Text Watermarking demo')

watermark.py

@@ -8,9 +8,6 @@ from scipy.special import gamma, gammainc, gammaincc, betainc
 from scipy.optimize import fminbound
 import numpy as np
 
-import os
-
-hf_token = os.getenv('HF_TOKEN')
 
 
 device = torch.device('cuda:0') if torch.cuda.is_available() else torch.device('cpu')