Update app.py

app.py

@@ -14,13 +14,21 @@ 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.strip()}\nCategories of the text:"""
-    results_dirty = llm.invoke(prompt)
+    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.
+    Text: {text.strip()}
+    Categories:"""
+    
+    # Invoke the model with the refined prompt
+    results = llm.invoke(prompt)
+    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:"""
+    #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."""
-    results_clean = llm.invoke(clean_prompt)
-    return results_clean
+    #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
 
 # Create Gradio interface
 interface = gr.Interface(