Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
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
|
| 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]. "
|