Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -53,21 +53,26 @@ def predict_text_api(payload: dict):
|
|
| 53 |
@api.post("/run/predict_image")
|
| 54 |
async def predict_image_api(file: UploadFile = File(...), explain_mode: str = Form(...)):
|
| 55 |
try:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 56 |
img_bytes = await file.read()
|
| 57 |
if not img_bytes:
|
| 58 |
-
raise ValueError("
|
| 59 |
-
|
|
|
|
| 60 |
|
| 61 |
return {
|
| 62 |
-
"
|
| 63 |
-
|
| 64 |
-
|
| 65 |
-
", ".join(result["suspicious_keywords"])
|
| 66 |
-
]
|
| 67 |
}
|
| 68 |
|
| 69 |
except Exception as e:
|
| 70 |
return {"error": str(e)}
|
|
|
|
| 71 |
# ✅ Gradio UI 功能
|
| 72 |
def predict_text(text, mode):
|
| 73 |
result = analyze_text(text=text, explain_mode=mode)
|
|
|
|
| 53 |
@api.post("/run/predict_image")
|
| 54 |
async def predict_image_api(file: UploadFile = File(...), explain_mode: str = Form(...)):
|
| 55 |
try:
|
| 56 |
+
if not file:
|
| 57 |
+
raise ValueError("未上傳圖片")
|
| 58 |
+
if not explain_mode:
|
| 59 |
+
raise ValueError("未指定分析模式")
|
| 60 |
+
|
| 61 |
img_bytes = await file.read()
|
| 62 |
if not img_bytes:
|
| 63 |
+
raise ValueError("圖片內容為空")
|
| 64 |
+
|
| 65 |
+
result = analyze_image(img_bytes, explain_mode=explain_mode)
|
| 66 |
|
| 67 |
return {
|
| 68 |
+
"status": result["status"],
|
| 69 |
+
"confidence": f'{result["confidence"]}%',
|
| 70 |
+
"suspicious_keywords": result["suspicious_keywords"]
|
|
|
|
|
|
|
| 71 |
}
|
| 72 |
|
| 73 |
except Exception as e:
|
| 74 |
return {"error": str(e)}
|
| 75 |
+
|
| 76 |
# ✅ Gradio UI 功能
|
| 77 |
def predict_text(text, mode):
|
| 78 |
result = analyze_text(text=text, explain_mode=mode)
|