Update app.py

app.py

@@ -18,10 +18,10 @@ def predict_and_draw(input_image):
     #input_image=ImageToCV(input_image)
     
     instructions = model.predict(input_image)
-    img = cv2.imread(input_image, cv2.IMREAD_GRAYSCALE)
+    img = cv2.imread(input_image)
     img = cv2.resize(img, (IMAGE_SIZE, IMAGE_SIZE))
     # 創建一個新的白色圖像
-    image = np.zeros((IMAGE_SIZE, IMAGE_SIZE), dtype=np.uint8)
+    image = np.zeros((IMAGE_SIZE, IMAGE_SIZE, 3), dtype=np.uint8)
     
     # 執行每條繪圖指令
     for instruction in instructions:
@@ -30,13 +30,13 @@ def predict_and_draw(input_image):
             import traceback
             traceback.print_exc()
     #image=CVtoImage(image)
-    return img, image, "\n".join(instructions)
+    return CVtoImage(img), CVtoImage(image), "\n".join(instructions)
 
 
 iface = gr.Interface(
     fn=predict_and_draw,
     inputs=gr.Image(type="filepath"),
-    outputs=[  gr.Image(type="numpy" ,image_mode="L" ,label="Input Image"), gr.Image(type="numpy" ,image_mode="L" ,label="Output Image"), gr.Textbox(label="Generated Instructions" ,show_copy_button=True)],
+    outputs=[  gr.Image(label="Input Image"), gr.Image(label="Output Image"), gr.Textbox(label="Generated Instructions" ,show_copy_button=True)],
     title="Image to Drawing Instructions",
     description="Upload an image, and the model will predict drawing instructions based on it."
 )