Maybe better clouds

app.py

@@ -27,12 +27,13 @@ def generate(*args):
     model_path = f"{filename}.obj"
     pc.save_obj(model_path)
     # model_path = None  # TODO
-    return [video_path, model_path]
+    return [video_path, model_path, model_path]
 
 
 def main():
     with gr.Blocks() as ui:
         gr.Markdown("# Pulsar+CLIP")
+        gr.Markdown("<a target="_blank" href="https://colab.research.google.com/github/https://colab.research.google.com/drive/1IvV3HGoNjRoyAKIX-aqSWa-t70PW3nPs"> <img src='https://colab.research.google.com/assets/colab-badge.svg' alt='Open In Colab'/></a>"
         gr.Markdown("Generate 3D point clouds from text!")
         
         with gr.Group():
@@ -63,14 +64,15 @@ def main():
         gr.Markdown("## Result")
         with gr.Row():
             with gr.Column():
-                video = gr.Video()
+                video_result = gr.Video()
             with gr.Column():
-                model = gr.Model3D()
+                model_demo = gr.Model3D()
+                model_file = gr.File()
         
-        button.click(fn=generate, inputs=inputs, outputs=[video, model])
+        button.click(fn=generate, inputs=inputs, outputs=[video_result, model_demo, model_file])
         
         gr.Markdown("## Examples")
-        gr.Examples(fn=generate, inputs=inputs, outputs=[video, model],
+        gr.Examples(fn=generate, inputs=inputs, outputs=[video_result, model_demo, model_file],
                     examples=[defaults], cache_examples=True, examples_per_page=1)
     return ui
 

pulsar_clip.py

@@ -28,7 +28,7 @@ CONFIG_SPEC = [
         ("offset_random", 0.75, float),
         ("xyz_random", 0.25, float),
         ("altitude_range", 0.3, float),
-        ("augments", 4, int),
+        ("augments", 3, int),
     ]),
     ("Optimization", [
         ("epochs", 6, int),
@@ -50,7 +50,7 @@ CONFIG_SPEC = [
         ("log_opacity", False, bool),      #@param {type: "boolean"}
 
         ("min_radius", 0.030, float),
-        ("max_radius", 0.170, float),
+        ("max_radius", 0.150, float),
         ("log_radius", False, bool),
 
         # TODO dynamically decide bezier_res