Sharathhebbar24
commited on
Commit
•
106a2ee
1
Parent(s):
783dbaa
Update app.py
Browse files
app.py
CHANGED
@@ -84,20 +84,21 @@ class LLM_Langchain():
|
|
84 |
|
85 |
def generate_response(self, input_text, context):
|
86 |
|
87 |
-
template = f"<|system|>\nYou are a intelligent chatbot.</s>\n<|user|>\n{input_text}.\n<|assistant|>"
|
88 |
llm = HuggingFaceHub(
|
89 |
repo_id = self.model_name,
|
90 |
model_kwargs = self.model_kwargs
|
91 |
)
|
92 |
-
llm_chain = LLMChain(
|
93 |
-
|
94 |
-
|
95 |
-
|
96 |
-
|
97 |
-
result = llm_chain.run({
|
98 |
-
|
99 |
-
|
100 |
-
})
|
|
|
101 |
# return llm(input_text)
|
102 |
return result
|
103 |
|
|
|
84 |
|
85 |
def generate_response(self, input_text, context):
|
86 |
|
87 |
+
template = f"<|system|>\nYou are a intelligent chatbot and expertise in {context}.</s>\n<|user|>\n{input_text}.\n<|assistant|>"
|
88 |
llm = HuggingFaceHub(
|
89 |
repo_id = self.model_name,
|
90 |
model_kwargs = self.model_kwargs
|
91 |
)
|
92 |
+
# llm_chain = LLMChain(
|
93 |
+
# prompt=template,
|
94 |
+
# llm=llm,
|
95 |
+
# )
|
96 |
+
|
97 |
+
# result = llm_chain.run({
|
98 |
+
# "question": input_text,
|
99 |
+
# "context": context
|
100 |
+
# })
|
101 |
+
result = llm(template)
|
102 |
# return llm(input_text)
|
103 |
return result
|
104 |
|