Update app.py

app.py

@@ -18,7 +18,7 @@ def predict(input_image):
     price = str(output)
 
     # 返回預測結果
-    return [input_image, price]
+    return [output, price]
 
 with gr.Blocks() as demo:
     # Title