Update app.py

app.py

@@ -14,9 +14,9 @@ llm = HuggingFaceEndpoint(
 )
 # Define the function to process user input
 def classify_text(text):
-    prompt = "Classify the following text into a category or topic:"
-    input_text = f"{prompt}\n{text}"
-    results = llm.invoke(input_text)
+    prompt = f"""Classify the following text into a category or topic. You dont need to write specific informations or explanations, only return the categories. 
+    {text}\nCategories:"""
+    results = llm.invoke(prompt)
     return results
 
 # Create Gradio interface