neverix commited on
Commit
4fc543d
1 Parent(s): 15409f2

Make it work

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. pulsar_clip.py +3 -3
app.py CHANGED
@@ -65,7 +65,7 @@ def main():
65
  with gr.Column():
66
  video = gr.Video()
67
  with gr.Column():
68
- model = gr.Model3D()
69
 
70
  button.click(fn=generate, inputs=inputs, outputs=[video, model])
71
 
 
65
  with gr.Column():
66
  video = gr.Video()
67
  with gr.Column():
68
+ model = gr.File()
69
 
70
  button.click(fn=generate, inputs=inputs, outputs=[video, model])
71
 
pulsar_clip.py CHANGED
@@ -31,8 +31,8 @@ CONFIG_SPEC = [
31
  ("augments", 4, int),
32
  ]),
33
  ("Optimization", [
34
- ("epochs", 1, int),
35
- ("lr", 0.5, float),
36
  #@markdown CLIP loss type, might improve the results
37
  ("loss_type", "spherical", ["spherical", "cosine"]),
38
  #@markdown CLIP loss weight
@@ -50,7 +50,7 @@ CONFIG_SPEC = [
50
  ("log_opacity", False, bool), #@param {type: "boolean"}
51
 
52
  ("min_radius", 0.030, float),
53
- ("max_radius", 0.070, float),
54
  ("log_radius", False, bool),
55
 
56
  # TODO dynamically decide bezier_res
 
31
  ("augments", 4, int),
32
  ]),
33
  ("Optimization", [
34
+ ("epochs", 6, int),
35
+ ("lr", 0.6, float),
36
  #@markdown CLIP loss type, might improve the results
37
  ("loss_type", "spherical", ["spherical", "cosine"]),
38
  #@markdown CLIP loss weight
 
50
  ("log_opacity", False, bool), #@param {type: "boolean"}
51
 
52
  ("min_radius", 0.030, float),
53
+ ("max_radius", 0.170, float),
54
  ("log_radius", False, bool),
55
 
56
  # TODO dynamically decide bezier_res