a

app.py

@@ -29,7 +29,7 @@ def generate(image):
     
     return generated_caption, vega_image
 
-def draw_vega(vega, scale=1):
+def draw_vega(vega, scale=3):
 
     spec = json.dumps(vega, indent=4)
     png_data = vlc.vegalite_to_png(vl_spec=spec, scale=scale)
@@ -56,7 +56,10 @@ iface = gr.Interface(
 
     fn=generate,
     inputs=gr.Image(type="pil"),
-    outputs=[gr.Textbox(), gr.Image(type="pil")],
+
+    outputs=[gr.Textbox(),
+             gr.Image(type="pil", label="Generated Vega Image", tool="editor", shape=None).style(height=500, width=500)],
+             
     title="Image to Vega-Lite",
     description="Upload an image to generate vega-lite"
 )