Update app.py

app.py

@@ -12,7 +12,8 @@ class BasicAgent:
     def __init__(self):
         """用 HuggingFace Inference API"""
         self.hf_token = os.getenv("HF_TOKEN")
-        self.model_name = "meta-llama/Llama-2-7b-chat-hf"
+        # 改用更穩定的小模型
+        self.model_name = "gpt2"  # 或試試 "google/flan-t5-base"
         
         if not self.hf_token:
             print("✗ HF_TOKEN not found")
@@ -21,7 +22,7 @@ class BasicAgent:
         
         self.agent_url = f"{HF_API_URL}/{self.model_name}"
         self.agent = True
-        print("✓ Agent initialized with HF API")
+        print(f"✓ Agent initialized with model: {self.model_name}")
 
     def __call__(self, question: str) -> str:
         """透過 HF Inference API 呼叫模型"""
@@ -35,8 +36,6 @@ class BasicAgent:
                 "inputs": question,
                 "parameters": {
                     "max_new_tokens": 256,
-                    "temperature": 0.7,
-                    "top_p": 0.9
                 }
             }
             
@@ -47,17 +46,20 @@ class BasicAgent:
                 timeout=60
             )
             
+            if response.status_code == 410:
+                return "Model loading... please try again"
+            
             if response.status_code != 200:
-                return f"API Error {response.status_code}"
+                return f"API Error {response.status_code}: {response.text[:100]}"
             
             result = response.json()
             
             if isinstance(result, list) and len(result) > 0:
                 answer = result[0].get("generated_text", "")
                 answer = answer.replace(question, "").strip()
-                return answer[:1000] if answer else "No answer"
+                return answer[:1000] if answer else "No answer generated"
             
-            return "Invalid response"
+            return "Invalid response format"
             
         except Exception as e:
             return f"Error: {str(e)[:100]}"
@@ -70,8 +72,6 @@ def run_and_submit_all(profile: Optional[gr.OAuthProfile] = None):
         return "Please Login to Hugging Face with the button.", None
     
     username = profile.username
-    
-    # 自動抓 SPACE_ID（不用手動設定）
     space_id = os.getenv("SPACE_ID", "s1144662")
 
     api_url = DEFAULT_API_URL