Update app.py

app.py

@@ -1,5 +1,17 @@
+import os
+from dotenv import load_dotenv
 import gradio as gr
 from transformers import pipeline
+from huggingface_hub import login
+
+load_dotenv()
+
+hf_token = os.getenv('HUGGING_FACE_TOKEN')
+
+if hf_token:
+    login(token=hf_token)
+else:
+    raise ValueError("HUGGING_FACE_TOKEN is not set in the environment variables")
 
 def classify_text(text):
     pipe = pipeline("text-classification", model="meta-llama/Prompt-Guard-86M")
@@ -13,4 +25,5 @@ demo = gr.Interface(
     title="Prompt-Guard-86M Text Classification"
 )
 
-demo.launch(server_name="0.0.0.0", server_port=7860)
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)