s1144662 commited on
Commit
287b0f6
·
verified ·
1 Parent(s): 7e35574

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -13,7 +13,8 @@ class BasicAgent:
13
  """用 HuggingFace Inference API"""
14
  self.hf_token = os.getenv("HF_TOKEN")
15
  # 改用更穩定的小模型
16
- self.model_name = "gpt2" # 或試試 "google/flan-t5-base"
 
17
 
18
  if not self.hf_token:
19
  print("✗ HF_TOKEN not found")
 
13
  """用 HuggingFace Inference API"""
14
  self.hf_token = os.getenv("HF_TOKEN")
15
  # 改用更穩定的小模型
16
+ self.model_name = "google/flan-t5-base"
17
+
18
 
19
  if not self.hf_token:
20
  print("✗ HF_TOKEN not found")