carlosrosas commited on
Commit
69ce27b
·
verified ·
1 Parent(s): efc0c0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -15,7 +15,7 @@ import pandas as pd
15
  device = "cuda" if torch.cuda.is_available() else "cpu"
16
 
17
  # Define variables
18
- temperature = 0.7
19
  max_new_tokens = 3000
20
  top_p = 0.95
21
  repetition_penalty = 1.2
@@ -26,8 +26,8 @@ model_name = "PleIAs/Cassandre-RAG"
26
  llm = LLM(model_name, max_model_len=8128)
27
 
28
  # Connect to the LanceDB database
29
- db = lancedb.connect("content19/lancedb_data")
30
- table = db.open_table("edunat19")
31
 
32
  def hybrid_search(text):
33
  results = table.search(text, query_type="hybrid").limit(5).to_pandas()
 
15
  device = "cuda" if torch.cuda.is_available() else "cpu"
16
 
17
  # Define variables
18
+ temperature = 0.6
19
  max_new_tokens = 3000
20
  top_p = 0.95
21
  repetition_penalty = 1.2
 
26
  llm = LLM(model_name, max_model_len=8128)
27
 
28
  # Connect to the LanceDB database
29
+ db = lancedb.connect("abstracts/lancedb_data")
30
+ table = db.open_table("abstractsC")
31
 
32
  def hybrid_search(text):
33
  results = table.search(text, query_type="hybrid").limit(5).to_pandas()