Fix bug in code cleaning

app.py

@@ -34,14 +34,10 @@ ANSWER:\
 
 def clean_text(generation):
     # clean up text has discussed in OpenAI's paper "Evaluating Large Language Models Trained on Code"
-    generation = generation.split("\
-def")[0]
-    generation = generation.split("\
-class")[0]
-    generation = generation.split("\
-#")[0]
-    generation = generation.split("\
-if")[0]
+    generation = generation.split("\ndef")[0]
+    generation = generation.split("\nclass")[0]
+    generation = generation.split("\n#")[0]
+    generation = generation.split("\nif")[0]
 
     return generation