add hub login

requirements.txt

@@ -2,4 +2,5 @@ bs4==0.0.1
 inflect==7.0.0
 openai==0.27.7
 pandas==1.5.3
-langchain==0.0.327
+langchain==0.0.327
+huggingface-hub==0.16.4

src/few_shot_funcs.py

@@ -5,12 +5,15 @@ import inflect
 import pandas as pd
 from typing import Dict
 from datasets import load_dataset
+from huggingface_hub import login
 from langchain.embeddings.openai import OpenAIEmbeddings
 from langchain.vectorstores import FAISS
 from langchain.vectorstores.utils import DistanceStrategy
 
 
+# Get OpenAI and huggingface-hub keys
 openai.api_key = os.environ.get('OPENAI_KEY')
+login(os.environ.get('HUB_KEY'))
 
 
 # Constants
@@ -18,7 +21,7 @@ FS_COLUMNS = ['asin', 'category', 'title', 'tech_process', 'labels']
 MAX_TOKENS = 700
 USER_TXT = 'Write feature-bullets for an Amazon product page. ' \
            'Title: {title}. Technical details: {tech_data}.\n\n### Feature-bullets:'
-           
+
 # Load few-shot dataset
 FS_DATASET = load_dataset('iarbel/amazon-product-data-filter', split='validation')