IamVicky111 commited on
Commit
6cdcd01
·
verified ·
1 Parent(s): 71dba73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -6,6 +6,9 @@ from langchain_community.llms import HuggingFaceEndpoint
6
  from langchain_community.embeddings import HuggingFaceInferenceAPIEmbeddings
7
  import gradio as gr
8
  import subprocess
 
 
 
9
  #Using Mistral Modal
10
  # Ensure Playwright installs required browsers and dependencies
11
  subprocess.run(["playwright", "install"])
@@ -25,9 +28,15 @@ embedder_model_instance = HuggingFaceInferenceAPIEmbeddings(
25
  api_key=HUGGINGFACEHUB_API_TOKEN, model_name="sentence-transformers/all-MiniLM-l6-v2"
26
  )
27
 
 
 
 
 
28
  graph_config = {
29
  "llm": {"model_instance": llm_model_instance},
30
- "embeddings": {"model_instance": embedder_model_instance}
 
 
31
  }
32
  #To Scrape the data and summarize it
33
  def scrape_and_summarize(prompt, source):
 
6
  from langchain_community.embeddings import HuggingFaceInferenceAPIEmbeddings
7
  import gradio as gr
8
  import subprocess
9
+ import redis
10
+ from langchain_community.vectorstores.redis import RedisVectorStore
11
+
12
  #Using Mistral Modal
13
  # Ensure Playwright installs required browsers and dependencies
14
  subprocess.run(["playwright", "install"])
 
28
  api_key=HUGGINGFACEHUB_API_TOKEN, model_name="sentence-transformers/all-MiniLM-l6-v2"
29
  )
30
 
31
+
32
+ r = redis.Redis(host="localhost", port=6379)
33
+ vector_store = RedisVectorStore(redis=r)
34
+
35
  graph_config = {
36
  "llm": {"model_instance": llm_model_instance},
37
+ "embeddings": {"model_instance": embedder_model_instance},
38
+ "vector_store": {"model_instance": vector_store}
39
+ }
40
  }
41
  #To Scrape the data and summarize it
42
  def scrape_and_summarize(prompt, source):