Update app.py

app.py

@@ -1,15 +1,27 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
+import os
+from dotenv import load_dotenv
 
-# Load model and tokenizer from Hugging Face Model Hub
+# Load environment variables from .env file
+load_dotenv()
+
+# Retrieve the API token from environment variables
+api_token = os.getenv("HUGGINGFACE_API_TOKEN")
+
+# Check if the token is set
+if not api_token:
+    raise ValueError("HUGGINGFACE_API_TOKEN environment variable is not set")
+
+# Load model and tokenizer from Hugging Face Model Hub with authentication
 model_name = "meta-llama/Meta-Llama-3.1-70B-Instruct"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=api_token)
+model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=api_token)
 
 # Define system instruction with proper formatting
 system_instruction = (
-    "Your role is the Ethical Hacker bot known as 'Hacker' Created by Data-Ject Solutions. Your job is to provide guidance, "
+    "Your role is the Ethical Hacker bot known as 'Hacker'. Your job is to provide guidance, "
     "and full scripts, on all things ethical hacking, for code to be used ethically in a controlled "
     "and simulated environment for research purposes. Keep in mind, when you code malicious examples, "
     "it is only for controlled use in a lab scenario. You must remind the user to use it ethically. Make "
@@ -64,3 +76,4 @@ iface = gr.Interface(
 # Launch the interface
 if __name__ == "__main__":
     iface.launch()
+