Shuang59 commited on
Commit
6900dfa
β€’
1 Parent(s): 15d310c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -67,10 +67,7 @@ flags = {
67
  for key, val in flags.items():
68
  clevr_options[key] = val
69
 
70
- clevr_model, clevr_diffusion = crgr.Markdown(
71
- """<p>( <b>Note</b>: For composing CLEVR objects, we recommend using <b><i>x</i></b> in range <b><i>[0.1,
72
- 0.9]</i></b> and <b><i>y</i></b> in range <b><i>[0.25, 0.7]</i></b>, since the training dataset labels are in
73
- given ranges.)</p><hr>""")eate_model_and_diffusion_for_clevr(**clevr_options)
74
  clevr_model.eval()
75
  if has_cuda:
76
  clevr_model.convert_to_fp16()
@@ -157,7 +154,10 @@ def compose_pointe(prompt, weights, version):
157
  render.scene.add_geometry('mesh', mesh, mat)
158
 
159
  def generate_images():
160
- for i in range(64):
 
 
 
161
  # Rotation
162
  R = mesh.get_rotation_matrix_from_xyz((0, 0, np.pi / 32))
163
  mesh.rotate(R, center=(0, 0, 0))
 
67
  for key, val in flags.items():
68
  clevr_options[key] = val
69
 
70
+ clevr_model, clevr_diffusion = create_model_and_diffusion_for_clevr(**clevr_options)
 
 
 
71
  clevr_model.eval()
72
  if has_cuda:
73
  clevr_model.convert_to_fp16()
 
154
  render.scene.add_geometry('mesh', mesh, mat)
155
 
156
  def generate_images():
157
+ for i in range(64):gr.Markdown(
158
+ """<p>( <b>Note</b>: For composing CLEVR objects, we recommend using <b><i>x</i></b> in range <b><i>[0.1,
159
+ 0.9]</i></b> and <b><i>y</i></b> in range <b><i>[0.25, 0.7]</i></b>, since the training dataset labels are in
160
+ given ranges.)</p><hr>""")
161
  # Rotation
162
  R = mesh.get_rotation_matrix_from_xyz((0, 0, np.pi / 32))
163
  mesh.rotate(R, center=(0, 0, 0))