Tanmay211998 commited on
Commit
e98a425
·
verified ·
1 Parent(s): f090855

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -6,6 +6,7 @@ from llama_index.llms.huggingface import HuggingFaceLLM
6
  from llama_index.core.prompts.prompts import SimpleInputPrompt
7
  import subprocess
8
  import os
 
9
 
10
  documents = SimpleDirectoryReader('./pdf').load_data()
11
  documents
@@ -21,6 +22,9 @@ query_wrapper_prompt = SimpleInputPrompt("<|USER|>{query_str}<|ASSISTANT|>")
21
  # Run the command to login using the token
22
  VarHFToken = os.getenv("HFToken")
23
  print(VarHFToken)
 
 
 
24
  #command = f"huggingface-cli login --token {VarHFToken}"
25
  #subprocess.run(command, shell=True)
26
 
 
6
  from llama_index.core.prompts.prompts import SimpleInputPrompt
7
  import subprocess
8
  import os
9
+ from huggingface_hub import login
10
 
11
  documents = SimpleDirectoryReader('./pdf').load_data()
12
  documents
 
22
  # Run the command to login using the token
23
  VarHFToken = os.getenv("HFToken")
24
  print(VarHFToken)
25
+ login(VarHFToken)
26
+
27
+
28
  #command = f"huggingface-cli login --token {VarHFToken}"
29
  #subprocess.run(command, shell=True)
30