Update app.py

app.py

@@ -4,11 +4,15 @@ import transformers
 from transformers import AutoTokenizer
 from langchain import LLMChain, HuggingFacePipeline, PromptTemplate
 import os
+from huggingface_hub import login
+
+
 
 access_token = os.getenv("Llama2")
 
-def greet(text):
+def greet(token, text):
 
+    login(token)
     model = "meta-llama/Llama-2-7b-chat-hf"
     tokenizer = AutoTokenizer.from_pretrained(model)
     
@@ -43,11 +47,12 @@ def greet(text):
     return summary
 
 with gr.Blocks() as demo:
-    
+
+    token = gr.Textbox(label=token)
     text = gr.Textbox(label="Text")
     summary = gr.Textbox(label="Summary")
     greet_btn = gr.Button("Submit")
-    greet_btn.click(fn=greet, inputs=text, outputs=summary, api_name="greet")
+    greet_btn.click(fn=greet, inputs=[token,text], outputs=summary, api_name="greet")
 
 
 demo.launch()