Spaces:
Running
Running
Update rag.py
Browse files
rag.py
CHANGED
@@ -50,11 +50,11 @@ class RAG:
|
|
50 |
}
|
51 |
|
52 |
query = f"### Instruction\n{instruction}\n\n### Context\n{context}\n\n### Answer\n "
|
53 |
-
prompt = "You are a helpful assistant. Answer the question using only the context you are provided with. If it is not possible to do it with the context, just say 'I can't answer'. <|endoftext|>"
|
54 |
|
55 |
|
56 |
payload = {
|
57 |
-
"inputs":
|
58 |
"parameters": {"MAX_NEW_TOKENS": 1000, "TEMPERATURE": 0.25}
|
59 |
}
|
60 |
|
|
|
50 |
}
|
51 |
|
52 |
query = f"### Instruction\n{instruction}\n\n### Context\n{context}\n\n### Answer\n "
|
53 |
+
#prompt = "You are a helpful assistant. Answer the question using only the context you are provided with. If it is not possible to do it with the context, just say 'I can't answer'. <|endoftext|>"
|
54 |
|
55 |
|
56 |
payload = {
|
57 |
+
"inputs": query,
|
58 |
"parameters": {"MAX_NEW_TOKENS": 1000, "TEMPERATURE": 0.25}
|
59 |
}
|
60 |
|