RalphThings commited on
Commit
584b7c0
·
verified ·
1 Parent(s): f925b4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -20,10 +20,8 @@ class BasicAgent:
20
  raise ValueError("HF_TOKEN not set in environment")
21
  self.generator = pipeline(
22
  "text-generation",
23
- model="gpt2", # or your chosen model
24
- tokenizer="gpt2",
25
  framework="pt",
26
- device=-1, # CPU; adjust if GPU
27
  use_auth_token=HF_TOKEN,
28
  )
29
  # The GAIA system prompt (no "FINAL ANSWER:" at the end)
 
20
  raise ValueError("HF_TOKEN not set in environment")
21
  self.generator = pipeline(
22
  "text-generation",
23
+ model="gpt2",
 
24
  framework="pt",
 
25
  use_auth_token=HF_TOKEN,
26
  )
27
  # The GAIA system prompt (no "FINAL ANSWER:" at the end)