Sharathhebbar24 commited on
Commit
106a2ee
1 Parent(s): 783dbaa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -10
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
- prompt=template,
94
- llm=llm,
95
- )
96
-
97
- result = llm_chain.run({
98
- "question": input_text,
99
- "context": context
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