Update app.py

app.py

@@ -57,12 +57,12 @@ def make_prompt(entry):
     return  f"### Human:{entry} ### Assistant:"
     # f"TELL A STORY, RELATE TO COMPUTER SCIENCE, INCLUDE ASSESMENTS. MAKE IT REALISTIC AND AROUND 800 WORDS, END THE STORY WITH "THE END.": {entry}"
 
-def process_text(text):
-    print("Before:", text)
-    processed_text = re.sub(r'\[.*?\]', '', text, flags=re.DOTALL)
-    print("After:", processed_text)
 
-    return processed_text
+def process_text(text):
+    text = re.sub(r'\[assessment;[^\]]*\]', '', text, flags=re.DOTALL)
+    text = re.sub(r'\[.*?\]', '', text, flags=re.DOTALL)
+    
+    return text
     
 @spaces.GPU
 def generate(