kk

app.py

@@ -57,9 +57,10 @@ def whisper_stt(audio):
 # LLM - Bloom Response  
 def lang_model_response(prompt): 
   print(f"*****Inside lang_model_response - Prompt is :{prompt}")
-  p = """If the following text is a question reply with an answer, else complete the sentence.
-  this is the text: """
-  prompt = p + prompt #+ "\n" #+ "Response: "
+  p = """Question: How are you doing today?
+  Answer: I am doing good, thanks.
+  Question: """
+  prompt = p + prompt + "\n" + "Answer: "
   #prompt = prompt + "\n"
   
   if len(prompt) == 0:
@@ -89,7 +90,7 @@ def lang_model_response(prompt):
   #elif '.' in prompt:
   #  solution = output_tmp.split(".")[1]
   #else:
-  solution = output_tmp.split("\n")[1]
+  solution = output_tmp.split("Answer:")[2]
     
   print(f"Final response after splits is: {solution}")
   return solution