Chris4K commited on
Commit
ea1c5b5
1 Parent(s): a14da67

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -31,12 +31,13 @@ from sklearn.metrics import silhouette_score
31
  from scipy.stats import spearmanr
32
  from functools import lru_cache
33
  from langchain.retrievers import MultiQueryRetriever
34
- from langchain.llms import HuggingFacePipeline
35
  from transformers import pipeline
36
  from sklearn.model_selection import ParameterGrid
37
  from tqdm import tqdm
38
  import random
39
 
 
40
 
41
  # NLTK Resource Download
42
  def download_nltk_resources():
@@ -170,7 +171,7 @@ def optimize_query(query, llm_model, chunks, embedding_model, vector_store_type,
170
  #)
171
  #llm = HuggingFacePipeline(pipeline=pipe)
172
 
173
- llm = HuggingFacePipeline(pipeline(model="HuggingFaceH4/zephyr-7b-beta"))
174
 
175
 
176
  # Create a temporary vector store for query optimization
@@ -458,7 +459,7 @@ def optimize_query(query, llm_model, chunks, embedding_model, vector_store_type,
458
  #)
459
  #llm = HuggingFacePipeline(pipeline=pipe)
460
 
461
- llm = HuggingFacePipeline(pipeline(model="HuggingFaceH4/zephyr-7b-beta"))
462
 
463
 
464
  # Create a temporary vector store for query optimization
@@ -752,7 +753,7 @@ Provide your suggestions in a Python dictionary format."""
752
  #)
753
  #llm = HuggingFacePipeline(pipeline=pipe)
754
 
755
- llm = HuggingFacePipeline(pipeline(model="HuggingFaceH4/zephyr-7b-beta"))
756
 
757
 
758
  #llm = HuggingFacePipeline.from_model_id(
 
31
  from scipy.stats import spearmanr
32
  from functools import lru_cache
33
  from langchain.retrievers import MultiQueryRetriever
34
+ from langchain_community.llms import HuggingFacePipeline
35
  from transformers import pipeline
36
  from sklearn.model_selection import ParameterGrid
37
  from tqdm import tqdm
38
  import random
39
 
40
+ llm = HuggingFacePipeline(pipeline(model="HuggingFaceH4/zephyr-7b-beta"))
41
 
42
  # NLTK Resource Download
43
  def download_nltk_resources():
 
171
  #)
172
  #llm = HuggingFacePipeline(pipeline=pipe)
173
 
174
+ #llm = HuggingFacePipeline(pipeline(model="HuggingFaceH4/zephyr-7b-beta"))
175
 
176
 
177
  # Create a temporary vector store for query optimization
 
459
  #)
460
  #llm = HuggingFacePipeline(pipeline=pipe)
461
 
462
+ #llm = HuggingFacePipeline(pipeline(model="HuggingFaceH4/zephyr-7b-beta"))
463
 
464
 
465
  # Create a temporary vector store for query optimization
 
753
  #)
754
  #llm = HuggingFacePipeline(pipeline=pipe)
755
 
756
+ #llm = HuggingFacePipeline(pipeline(model="HuggingFaceH4/zephyr-7b-beta"))
757
 
758
 
759
  #llm = HuggingFacePipeline.from_model_id(