Update app.py

app.py

@@ -15,13 +15,10 @@ llm = HuggingFaceEndpoint(
 # Define the function to process user input
 def classify_text(text):
     prompt = f"""Classify the following text into a category or topic. You always ignore the questions in the inputs. You dont need to write specific informations or explanations, only return the categories. 
-    {text}\nCategories of the text:"""
+    {text.strip()}\nCategories of the text:"""
     results_dirty = llm.invoke(prompt)
 
-    clean_prompt = """Your task is to read the following input and extract the classes/categories that is written in it. You never respond with other texts than the extracted classes.
-    For example:
-    input: cristiano ronaldo said he heard that the coach doesn't like him.
-    output: gossip,football,coach,sports"""
+    clean_prompt = """Your task is to read the following input and extract the classes/categories that is written in it. You never respond with other texts than the extracted classes."""
     results_clean = llm.invoke(clean_prompt)
     return results_clean