radames commited on
Commit
cbf69b2
·
1 Parent(s): ef67b7b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -9
app.py CHANGED
@@ -28,7 +28,7 @@ def enable_GPUS():
28
  print(d["name"])
29
 
30
 
31
- # enable_GPUS()
32
 
33
  # bpy.ops.wm.read_factory_settings(use_empty=True)
34
 
@@ -47,14 +47,14 @@ def generate(
47
  rgb1 = tuple(v / 255.0 for v in rgb1)
48
  rgb2 = ImageColor.getcolor(color2, "RGBA")
49
  rgb2 = tuple(v / 255.0 for v in rgb2)
50
- # bpy.ops.wm.read_homefile(use_empty=True)
51
 
52
 
53
  # Delete all mesh objects from the scene
54
- bpy.ops.object.select_all(action="DESELECT")
55
- bpy.ops.object.select_by_type(type="MESH")
56
- bpy.ops.object.delete()
57
-
 
58
  # Add a torus
59
  bpy.ops.mesh.primitive_torus_add(
60
  major_radius=1.5,
@@ -127,14 +127,16 @@ def generate(
127
  bpy.context.scene.render.resolution_x = 256
128
  bpy.context.scene.render.image_settings.file_format = "JPEG"
129
  bpy.context.scene.render.filepath = f.name
130
- bpy.context.scene.frame_current = bpy.context.scene.frame_end
131
 
132
- with tqdm(total=bpy.context.scene.frame_end) as pbar:
133
 
134
  def elapsed(dummy):
135
  pbar.update()
136
 
137
  bpy.app.handlers.render_stats.append(elapsed)
 
 
 
138
  # bpy.ops.render.render(animation=False, write_still=True)
139
  # bpy.ops.render.render(animation=False, write_still=True)
140
  bpy.ops.render.render(animation=False, write_still=True)
@@ -179,4 +181,4 @@ with gr.Blocks() as demo:
179
  )
180
 
181
  demo.queue()
182
- demo.launch(debug=True, inline=True)
 
28
  print(d["name"])
29
 
30
 
31
+ enable_GPUS()
32
 
33
  # bpy.ops.wm.read_factory_settings(use_empty=True)
34
 
 
47
  rgb1 = tuple(v / 255.0 for v in rgb1)
48
  rgb2 = ImageColor.getcolor(color2, "RGBA")
49
  rgb2 = tuple(v / 255.0 for v in rgb2)
 
50
 
51
 
52
  # Delete all mesh objects from the scene
53
+ for obj in bpy.context.scene.objects:
54
+ # If the object is of MESH type
55
+ if obj.type == 'MESH':
56
+ # Delete the object
57
+ bpy.data.objects.remove(obj, do_unlink=True)
58
  # Add a torus
59
  bpy.ops.mesh.primitive_torus_add(
60
  major_radius=1.5,
 
127
  bpy.context.scene.render.resolution_x = 256
128
  bpy.context.scene.render.image_settings.file_format = "JPEG"
129
  bpy.context.scene.render.filepath = f.name
 
130
 
131
+ with tqdm() as pbar:
132
 
133
  def elapsed(dummy):
134
  pbar.update()
135
 
136
  bpy.app.handlers.render_stats.append(elapsed)
137
+ bpy.context.scene.frame_set(1)
138
+ bpy.context.scene.frame_current = 1
139
+
140
  # bpy.ops.render.render(animation=False, write_still=True)
141
  # bpy.ops.render.render(animation=False, write_still=True)
142
  bpy.ops.render.render(animation=False, write_still=True)
 
181
  )
182
 
183
  demo.queue()
184
+ demo.launch(debug=True, inline=True)