Update app.py

app.py

@@ -7,16 +7,17 @@ from langchain_huggingface import HuggingFaceEndpoint
 # Initialize the LLM and other components
 llm = HuggingFaceEndpoint(
     repo_id="mistralai/Mistral-7B-Instruct-v0.3",
-    task="text-classification",
+    task="text-generation",
     max_new_tokens=64,
     temperature=0.5,
     do_sample=False,
 )
 # Define the function to process user input
 def classify_text(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.
+    prompt = f"""Your mission is to classify the following text into a category or topic.
     Text: {text.strip()}
-    Categories:"""
+    You dont need to write specific informations or explanations, only return the categories.
+    """
     
     # Invoke the model with the refined prompt
     results = llm.invoke(prompt).strip()