a

app.py

@@ -27,14 +27,14 @@ def generate(image):
     vega = string_to_vega(generated_caption)
     vega_image = draw_vega(vega)
     
-    return generated_caption, vega_image.thumbnail((500,500))
+    return generated_caption, vega_image
 
 def draw_vega(vega, scale=3):
 
     spec = json.dumps(vega, indent=4)
     png_data = vlc.vegalite_to_png(vl_spec=spec, scale=scale)
 
-    return Image.open(BytesIO(png_data))
+    return Image.open(BytesIO(png_data)).thumbnail((500,500))
 
 def string_to_vega(string):