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()