more debugging

app.py

@@ -5,6 +5,7 @@ from dataclasses import fields
 import gradio as gr
 import transformers
 from legogpt.models import LegoGPT, LegoGPTConfig
+from legogpt.data import LegoStructure
 import zipfile
 
 
@@ -53,7 +54,7 @@ def main():
         # Generate LEGO
         print(f'Generating LEGO for prompt: "{prompt}"')
         # output = model(prompt)
-        output = '2x4 (0,0,0)\n'
+        output = {'lego': LegoStructure.from_txt('2x4 (0,0,0)\n')}
         print(f'Finished generation!')
 
         # Render results and write to files