Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -29,6 +29,8 @@ model_name = "inceptionai/jais-13b"
|
|
29 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
30 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
31 |
|
|
|
|
|
32 |
# RAG function
|
33 |
def get_response(query, k=3):
|
34 |
query_embedding = embedding_model.encode([query])
|
|
|
29 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
30 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
31 |
|
32 |
+
|
33 |
+
|
34 |
# RAG function
|
35 |
def get_response(query, k=3):
|
36 |
query_embedding = embedding_model.encode([query])
|