Update app.py

app.py

@@ -19,14 +19,21 @@ tokenizer.set_chat_template(tokenizer.chat_template)
 
 
 def generate_response(prompt):
-    generated = pipe.generate([prompt], max_length=1024)
-    tokenpersec=f'{generated.perf_metrics.get_throughput().mean:.2f}'
-    match = re.search(r"<think>(.*?)</think>(.*)", generated, re.DOTALL)
+    try:
+        generated = pipe.generate([prompt], max_length=1024)
+        tokenpersec=f'{generated.perf_metrics.get_throughput().mean:.2f}'
+        match = re.search(r"<think>(.*?)</think>(.*)", generated, re.DOTALL)
     
-    if match:
-        thinking = match.group(1).strip()
-        content = match.group(2).strip()  
-    return tokenpersec, thinking, content
+        if match:
+            thinking = match.group(1).strip()
+            content = match.group(2).strip()
+        else:
+            thinking = "模型沒有提供思考過程"
+            content = generated  # 或者 generated.text， 取決於 generated 物件的屬性
+    
+        return tokenpersec, thinking, content
+    except Exception as e:
+        return "發生錯誤", "發生錯誤", f"生成回應時發生錯誤：{e}"
     
 
 # 建立 Gradio 介面