Update app.py

app.py

@@ -1,6 +1,16 @@
 import gradio as gr
 from transformers import LlamaTokenizer, LlamaForCausalLM
 import torch
+import os
+from huggingface_hub import login
+
+# Authenticate with Hugging Face using the token from Space secrets
+hf_token = os.getenv("HF_TOKEN")
+if hf_token:
+    login(token=hf_token)
+    print("Authenticated with Hugging Face token.")
+else:
+    print("HF_TOKEN not found in environment variables. Please set it in Space secrets.")
 
 # Function to predict fraud based on text input
 def predict(input_text):
@@ -8,7 +18,7 @@ def predict(input_text):
         return "Please enter some text to analyze."
     
     try:
-        # Load the fine-tuned model and tokenizer inside the function
+        # Load the fine-tuned model and tokenizer from the local directory
         tokenizer = LlamaTokenizer.from_pretrained("./fine_tuned_llama2")
         model = LlamaForCausalLM.from_pretrained("./fine_tuned_llama2")
         model.eval()