crodri commited on
Commit
47523db
1 Parent(s): 15c94ad

Update rag.py

Browse files
Files changed (1) hide show
  1. rag.py +2 -2
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": prompt + query,
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