Update app.py

app.py

@@ -3,8 +3,7 @@ from bert_explainer import analyze_text, analyze_image
 from fastapi import FastAPI, UploadFile, File, Form
 from fastapi.middleware.cors import CORSMiddleware
 import uvicorn
-import tempfile
-import os
+
 
 
 # ✅ 初始化 FastAPI
@@ -54,8 +53,10 @@ def predict_text_api(payload: dict):
 @api.post("/run/predict_image")
 async def predict_image_api(file: UploadFile = File(...), explain_mode: str = Form(...)):
     try:
-        img = await file.read()
-        result = analyze_image(img,explain_mode=explain_mode)
+        img_bytes = await file.read()
+        if not img_bytes:
+            raise ValueError("未收到圖片資料")
+        result = analyze_image(img_bytes, explain_mode = explain_mode)
 
         return {
             "data": [