Update app.py

app.py

@@ -18,11 +18,12 @@ def classify_text(text):
     {text}\nCategories of the text:"""
     results_dirty = llm.invoke(prompt)
 
-    clean = """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.
+    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"""
-    return results
+    results_clean = llm.invoke(clean_prompt)
+    return results_clean
 
 # Create Gradio interface
 interface = gr.Interface(