tdurzynski commited on
Commit
6daaf67
·
verified ·
1 Parent(s): 42866ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -27,8 +27,11 @@ logger = logging.getLogger(__name__)
27
  # Authenticate with Hugging Face
28
  login(HUGGING_API_KEY)
29
 
30
- # Load models and embeddings
31
- embedding_model = HuggingFaceHubEmbeddings(huggingfacehub_api_token=HUGGING_API_KEY)
 
 
 
32
  llm = ChatGroq(temperature=0, model_name="llama3-70b-8192", api_key=GROQ_API_KEY)
33
  client = AsyncGroq(api_key=GROQ_API_KEY)
34
 
 
27
  # Authenticate with Hugging Face
28
  login(HUGGING_API_KEY)
29
 
30
+ # Load models and embeddings with a specific embedding model
31
+ embedding_model = HuggingFaceHubEmbeddings(
32
+ huggingfacehub_api_token=HUGGING_API_KEY,
33
+ model_id="sentence-transformers/all-MiniLM-L6-v2" # Explicitly set a proper embedding model
34
+ )
35
  llm = ChatGroq(temperature=0, model_name="llama3-70b-8192", api_key=GROQ_API_KEY)
36
  client = AsyncGroq(api_key=GROQ_API_KEY)
37