Update app.py

app.py

@@ -16,7 +16,7 @@ tokenizer.pad_token = tokenizer.eos_token
 
 MAX_INPUT_TOKEN_LENGTH = 10000
 
-def generate_response(input_text, temperature=0.7, max_new_tokens=100):
+def generate_response(input_text, temperature=0.5, max_new_tokens=100):
     input_ids = tokenizer.encode(input_text, return_tensors='pt').to(model.device)
 
     if input_ids.shape[1] > MAX_INPUT_TOKEN_LENGTH:
@@ -65,7 +65,7 @@ def main():
             st.write("Archivo CSV cargado exitosamente:")
             st.write(df.head())
 
-            initial_prompt = f"The following is a list of job titles: {job_titles}. Please provide the first job title from this list."
+            initial_prompt = f"I have a list of job titles: {job_titles}. Please extract and return only the first job title from this list without repeating."
             st.write(f"Query: {query}")
             st.write(f"Prompt inicial: {initial_prompt}")