k

app.py

@@ -82,15 +82,10 @@ def lang_model_response(prompt):
   response = requests.post(API_URL, headers=headers, json=json_)
   print(f"Response  is : {response}")
   output = response.json()
-  print(f"output is : {output}") 
+  #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("Answer:")[2]
+  solution = output_tmp.split("Answer: ")[2].split("\n")[0]
     
   print(f"Final response after splits is: {solution}")
   return solution