Update app.py

app.py

@@ -121,7 +121,8 @@ top_margin = 100
 
 #out_image = put_text(improved_text, font, font_scale, color, thickness, max_width, out_image_width, top_margin)
 
-def predict(input_img):
+def predict(input_path):
+    image = cv.imread(input_path)
     bin_image = preprocess_image(input_img)
     lines = split_image_into_lines(bin_image)
     improved_text = get_improved_result(lines)
@@ -130,8 +131,8 @@ def predict(input_img):
 
 gradio_app = gr.Interface(
     predict,
-    inputs=gr.Image(label="Image with handwritten text", sources=['upload']),
-    outputs=[gr.Image(label="Output Image")],
+    inputs=gr.Image(sources=['upload', 'webcam'], type='filepath'),
+    outputs=[gr.Image()],
     title="Extract Handwritten Text",
 )