Update app.py

app.py

@@ -65,19 +65,15 @@ def main():
 
             # Definir el prompt con in-context learning
             initial_prompt = (
-                "Here are some examples of job title extraction:\n"
-                "Example 1:\n"
-                "List: ['Data Scientist', 'Machine Learning Engineer', 'AI Researcher']\n"
-                "First job title: 'Data Scientist'\n"
-                "\n"
-                "Example 2:\n"
-                "List: ['Software Developer', 'Backend Engineer', 'Frontend Developer']\n"
-                "First job title: 'Software Developer'\n"
-                "\n"
-                "Now, extract the first job title from the following list:\n"
+                "You are an AI assistant specialized in job title extraction and similarity calculation.\n"
+                "Your task is to extract the first job title from the following list and calculate the cosine similarity with the given query.\n"
                 f"List: {job_titles}\n"
-                "First job title:"
+                f"Query: '{query}'\n"
+                "Extracted first job title: \n"
+                "Cosine similarity score: "
             )
+
+
             
             st.write("Prompt inicial con In-context Learning:")
             st.write(initial_prompt)