Yhhxhfh commited on
Commit
b3714b6
·
verified ·
1 Parent(s): 0b5f8d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -5,11 +5,17 @@ from sklearn.feature_extraction.text import TfidfVectorizer
5
  from sklearn.metrics.pairwise import cosine_similarity
6
  import numpy as np
7
  from dotenv import load_dotenv
 
8
  import os
9
 
10
  load_dotenv()
11
  app = Flask(__name__)
12
 
 
 
 
 
 
13
  models = {
14
  "meta-llama": LLM(model="meta-llama/Meta-Llama-3.1-8B"),
15
  "gemma": LLM(model="google/gemma-2-9b"),
 
5
  from sklearn.metrics.pairwise import cosine_similarity
6
  import numpy as np
7
  from dotenv import load_dotenv
8
+ from huggingface_hub import snapshot_download, notebook_login, upload_folder, create_repo, login
9
  import os
10
 
11
  load_dotenv()
12
  app = Flask(__name__)
13
 
14
+ login(
15
+ token=os.getenv("HUGGINGFACE_TOKEN"),
16
+ add_to_git_credential=True
17
+ )
18
+
19
  models = {
20
  "meta-llama": LLM(model="meta-llama/Meta-Llama-3.1-8B"),
21
  "gemma": LLM(model="google/gemma-2-9b"),