Update app.py

app.py

@@ -1,5 +1,9 @@
 import gradio as gr
 
+import os
+
+api_key = os.getenv("TOKEN")
+
 def greet(name):
     return "Hello " + name + "!!"
 
@@ -12,7 +16,7 @@ from transformers import pipeline
 messages = [
     {"role": "user", "content": "Who are you?"},
 ]
-pipe = pipeline("text-generation", model="meta-llama/Llama-3.1-8B-Instruct",token=TOKEN)
+pipe = pipeline("text-generation", model="meta-llama/Llama-3.1-8B-Instruct",token=api_key)
 pipe(messages)