2

app.py

@@ -8,8 +8,8 @@ model_name = "rkwsuper/lora_model"
 # Use an environment variable or secret for the token
 auth_token = os.getenv("HF_TOKEN")  # Automatically fetches the token if set as an environment variable
 
-tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=auth_token)
-model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=auth_token)
+tokenizer = AutoTokenizer.from_pretrained(model_name, token=auth_token)
+model = AutoModelForCausalLM.from_pretrained(model_name, token=auth_token)
 
 # Define the function for inference
 def generate_text(prompt, max_length=100, temperature=1.0):