Spaces:
Paused
Paused
Update app.py
Browse files
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.
|
| 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("
|
| 30 |
-
table = db.open_table("
|
| 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()
|