Update app.py

app.py

@@ -14,13 +14,12 @@ llm = HuggingFaceEndpoint(
 )
 # Define the function to process user input
 def classify_text(text):
-    prompt = f"""Classify the following text into a category or topic. 
-    Ignore the questions in the input and only return the categories. Do not include any additional text or explanations.
+    prompt = f"""Classify the following text into relevant categories. Only provide category names, without any additional text, explanations, or details. 
     Text: {text.strip()}
     Categories:"""
     
     # Invoke the model with the refined prompt
-    results = llm.invoke(prompt)
+    results = llm.invoke(prompt).strip()
     return results
     #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.strip()}\nCategories of the text:"""