Update app.py

app.py

@@ -8,7 +8,7 @@ import os
 
 
 # Initialize the model pipeline
-generator = pipeline('text-generation', model='Locutusque/ChatHercules-2.5-Mistral-7B-DPO', torch_dtype=torch.bfloat16)
+generator = pipeline('text-generation', model='Locutusque/ChatHercules-2.5-Mistral-7B-DPO', torch_dtype=torch.bfloat16, os.environ["HF"])
 @spaces.GPU
 def generate_text(prompt, temperature, top_p, top_k, repetition_penalty, max_length):
     # Generate text using the model