Update app.py

app.py

@@ -1,9 +1,8 @@
-import os
 import numpy as np
-import cv2
 import base64
 import gradio as gr
 from tensorflow import keras
+import cv2
 
 # Load the model
 model_path = 'sketch2draw_model.h5'  # Update with your model path
@@ -33,7 +32,7 @@ with gr.Blocks() as demo:
     
     with gr.Row():
         with gr.Column():
-            canvas = gr.Sketchpad(shape=(400, 400), label="Draw Here", tool="brush")
+            canvas = gr.Sketchpad(label="Draw Here", tool="brush")
             brush_color = gr.ColorPicker(value="black", label="Brush Color")
             clear_btn = gr.Button("Clear")