FD900 commited on
Commit
c0652f3
·
verified ·
1 Parent(s): e52cce4

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -6,7 +6,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
6
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
7
  generator = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
8
 
9
- # GAIA system prompt (applied manually)
10
  system_prompt = (
11
  "You are a general AI assistant. I will ask you a question. Report your thoughts, "
12
  "and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. "
 
6
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
7
  generator = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
8
 
9
+ # GAIA system prompt
10
  system_prompt = (
11
  "You are a general AI assistant. I will ask you a question. Report your thoughts, "
12
  "and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. "