Revert "more debugging"

This reverts commit 133b3c70f0ff298e95fe193a6737ca7e326f3ed2.

app.py

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