Update app.py

app.py

@@ -31,6 +31,7 @@ def generate_response(input_text, temperature=0.2, max_new_tokens=100):
         do_sample=True,
         top_k=50,
         top_p=0.9,
+        num_beams=3,  # Beam search con 3 opciones
         temperature=temperature,
         eos_token_id=[tokenizer.eos_token_id]
     )
@@ -60,7 +61,7 @@ def main():
         job_titles = df['job_title'].tolist()
 
         # Prompt mejorado
-        initial_prompt = f"I have a list of job titles: {job_titles}. Please give me the first job title from the list."
+        initial_prompt = f"Here is a list of job titles: {job_titles}. Please only return the first job title from this list without any additional text."
         st.write(f"Query: {query}")
         st.write(f"Prompt inicial: {initial_prompt}")