Freddolin commited on
Commit
3fa4037
·
verified ·
1 Parent(s): b7ae39c

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -16,7 +16,7 @@ class GaiaAgent:
16
  # Använd en mer kapabel modell för bättre reasoning
17
  self.llm = pipeline(
18
  "text-generation",
19
- model="microsoft/DialoGPT-medium", # <-- Bara denna rad
20
  token=token,
21
  device="cpu",
22
  max_new_tokens=1024,
 
16
  # Använd en mer kapabel modell för bättre reasoning
17
  self.llm = pipeline(
18
  "text-generation",
19
+ model="google/flan-t5-large", # <-- Bara denna rad
20
  token=token,
21
  device="cpu",
22
  max_new_tokens=1024,