a

app.py

@@ -60,6 +60,7 @@ def lang_model_response(prompt):
   p = """Reply to this text in one short sentence.
   text: """
   #prompt = p + prompt + "\n" #+ "Response: "
+  prompt = prompt + "\n"
   
   if len(prompt) == 0:
     prompt = """Can you help me please?"""
@@ -83,12 +84,12 @@ def lang_model_response(prompt):
   print(f"output is : {output}") 
   output_tmp = output[0]['generated_text']
   print(f"output_tmp is: {output_tmp}")
-  if '?' in prompt:
-    solution = output_tmp.split("?")[1]
-  elif '.' in prompt:
-    solution = output_tmp.split(".")[1]
-  else:
-    solution = output_tmp.split("\n")[1]
+  #if '?' in prompt:
+  #  solution = output_tmp.split("?")[1]
+  #elif '.' in prompt:
+  #  solution = output_tmp.split(".")[1]
+  #else:
+  solution = output_tmp.split("\n")[1]
     
   print(f"Final response after splits is: {solution}")
   return solution