Update app.py

app.py

@@ -7,6 +7,7 @@ from resnest.torch import resnest50
 from rembg import remove
 from PIL import Image
 import io
+import requests
 
 # 加载类别名称
 with open('class_names.pkl', 'rb') as f:
@@ -77,6 +78,19 @@ def predict_image(img, remove_bg=False):
     # 记录结果
     best_class = class_names[top3_indices[0]]
     best_conf = top3_probs[0].item() * 100
+    
+
+    # 新增：调用本地API保存结果
+    api_url = "http://10.230.23.58:8806/save_result"  # 替换为你的本地IP
+    payload = {
+        "filename": "uploaded_image.jpg",  # 可改为实际文件名
+        "class": best_class,
+        "confidence": f"{best_conf:.2f}%"
+    }
+    try:
+        requests.post(api_url, json=payload, timeout=3)
+    except Exception as e:
+        print(f"保存到数据库失败: {e}")
 
     return processed_img, best_class, f"{best_conf:.2f}%", results