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"
+                "Step 1: Extract the first job title from the list.\n"
                 f"List: {job_titles}\n"
-                "First job title:"
+                "First job title: \n"
+                "\n"
+                "Step 2: Calculate the cosine similarity between the extracted job title and the given query.\n"
+                f"Query: '{query}'\n"
+                "Cosine similarity score: \n"
             )
+
             
             st.write("Prompt inicial con In-context Learning:")
             st.write(initial_prompt)