Jkalonji commited on
Commit
fb5d8fe
1 Parent(s): 19ca331

Adding a function to be used in gardio

Browse files
Files changed (1) hide show
  1. RAG.py +11 -1
RAG.py CHANGED
@@ -75,4 +75,14 @@ def generateResponseBasedOnContext(model_name, context_string, query):
75
  print("context : ", context)
76
 
77
  result = question_answerer(question=query, context=context)
78
- return result['answer']
 
 
 
 
 
 
 
 
 
 
 
75
  print("context : ", context)
76
 
77
  result = question_answerer(question=query, context=context)
78
+ return result['answer']
79
+
80
+
81
+ def gradio_adapted_RAG(model_name, query):
82
+ context = str(RAG_Langchain(query)['Content'])
83
+ generated_answer = generateResponseBasedOnContext(str(model_name),
84
+ context,
85
+ query)
86
+ return generated_answer
87
+
88
+