a

app.py

@@ -3,7 +3,8 @@ from transformers import AutoProcessor, Pix2StructForConditionalGeneration
 import torch
 from PIL import Image
 import json
-import vl_convert as vlc  # Ensure you have this library installed (pip install vl-convert)
+import vl_convert as vlc  
+from io import BytesIO
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -33,7 +34,7 @@ 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(png_data)
+    return Image.open(BytesIO(png_data))
 
 def string_to_vega(string):
 
@@ -62,4 +63,4 @@ iface = gr.Interface(
 
 # Launch the interface
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch(share=True)