ko

app.py

@@ -59,7 +59,7 @@ def lang_model_response(prompt):
   print(f"*****Inside lang_model_response - Prompt is :{prompt}")
   p = """Instruction: Reply to this Statement in one sentence.
   Statement: """
-  prompt = p + prompt + "\n" + "Response: "
+  prompt = p + prompt + "\n" #+ "Response: "
   
   if len(prompt) == 0:
     prompt = """Can you help me please?"""
@@ -83,7 +83,7 @@ def lang_model_response(prompt):
   print(f"output is : {output}") 
   output_tmp = output[0]['generated_text']
   print(f"output_tmp is: {output_tmp}")
-  solution = output_tmp.split("Response:")[1]   
+  solution = output_tmp.split("\n")[2]   
   print(f"Final response after splits is: {solution}")
   return solution