Upgrade to Gradio 4.36

README.md

@@ -4,7 +4,7 @@ emoji: 📚
 colorFrom: red
 colorTo: purple
 sdk: gradio
-sdk_version: 3.1.3
+sdk_version: 4.36.1
 python_version: 3.9.13
 app_file: app.py
 pinned: false

app.py

@@ -45,6 +45,8 @@ class App:
     def __call__(self, model_name, image):
         if image is None:
             return '', []
+        if isinstance(image, dict):  # Extact image from ImageEditor output
+            image = image['composite']
         model = self._get_model(model_name)
         image = self._preprocess(image.convert('RGB')).unsqueeze(0)
         # Greedy decoding
@@ -79,11 +81,11 @@ def main():
         model_name = gr.Radio(app.models, value=app.models[0], label='The STR model to use')
         with gr.Tabs():
             with gr.TabItem('Image Upload'):
-                image_upload = gr.Image(type='pil', source='upload', label='Image')
+                image_upload = gr.Image(type='pil', sources=['upload'], label='Image')
                 gr.Examples(glob.glob('demo_images/*.*'), inputs=image_upload)
                 read_upload = gr.Button('Read Text')
             with gr.TabItem('Canvas Sketch'):
-                image_canvas = gr.Image(type='pil', source='canvas', label='Sketch')
+                image_canvas = gr.ImageEditor(type='pil', sources=[], label='Sketch', image_mode='RGB', layers=False, canvas_size=(768, 192))
                 read_canvas = gr.Button('Read Text')
 
         output = gr.Textbox(max_lines=1, label='Model output')

requirements.txt

@@ -1,6 +1,6 @@
 --extra-index-url https://download.pytorch.org/whl/cpu
 
-gradio >=3.0.0, <4.0.0
+gradio ~=4.36.0
 torch >=2.0.0
 torchvision >=0.15.0
 pytorch-lightning ~=2.2.0