radames commited on
Commit
e8f9548
·
1 Parent(s): 3465760

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -14
app.py CHANGED
@@ -30,6 +30,8 @@ def enable_GPUS():
30
 
31
  enable_GPUS()
32
 
 
 
33
 
34
  def generate(
35
  color1,
@@ -121,23 +123,23 @@ def generate(
121
 
122
  # Render
123
  with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as f:
124
- bpy.context.scene.render.resolution_y = 256
125
- bpy.context.scene.render.resolution_x = 256
126
  bpy.context.scene.render.image_settings.file_format = "PNG"
127
  bpy.context.scene.render.filepath = f.name
128
 
129
  with tqdm(total=bpy.context.scene.frame_end) as pbar:
130
 
131
- def elapsed(dummy):
132
- pbar.update()
133
 
134
- bpy.app.handlers.render_stats.append(elapsed)
135
- bpy.ops.render.render(animation=False, write_still=True)
136
- bpy.data.images["Render Result"].save_render(
137
- filepath=bpy.context.scene.render.filepath
138
- )
139
- bpy.app.handlers.render_stats.clear()
140
- return f.name
141
 
142
 
143
  # generate("#ffffff", "#aaa", 1)
@@ -151,9 +153,7 @@ with gr.Blocks() as demo:
151
  camera_Z = gr.Slider(minimum=-100, maximum=100, value=4, label="Camera Z")
152
  torus_X = gr.Slider(minimum=-pi, maximum=pi, value=0, label="Torus φ")
153
  torus_Y = gr.Slider(minimum=-pi, maximum=pi, value=0, label="Torus θ")
154
- torus_Z = gr.Slider(
155
- minimum=-pi, maximum=2 * pi, value=pi / 2, label="Torus ψ"
156
- )
157
 
158
  render_btn = gr.Button("Render")
159
  with gr.Column(scale=3):
 
30
 
31
  enable_GPUS()
32
 
33
+ # bpy.ops.wm.read_factory_settings(use_empty=True)
34
+
35
 
36
  def generate(
37
  color1,
 
123
 
124
  # Render
125
  with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as f:
126
+ bpy.context.scene.render.resolution_y = 128
127
+ bpy.context.scene.render.resolution_x = 128
128
  bpy.context.scene.render.image_settings.file_format = "PNG"
129
  bpy.context.scene.render.filepath = f.name
130
 
131
  with tqdm(total=bpy.context.scene.frame_end) as pbar:
132
 
133
+ def elapsed(dummy):
134
+ pbar.update()
135
 
136
+ bpy.app.handlers.render_stats.append(elapsed)
137
+ bpy.ops.render.render(animation=False, write_still=True)
138
+ bpy.data.images["Render Result"].save_render(
139
+ filepath=bpy.context.scene.render.filepath
140
+ )
141
+ bpy.app.handlers.render_stats.clear()
142
+ return f.name
143
 
144
 
145
  # generate("#ffffff", "#aaa", 1)
 
153
  camera_Z = gr.Slider(minimum=-100, maximum=100, value=4, label="Camera Z")
154
  torus_X = gr.Slider(minimum=-pi, maximum=pi, value=0, label="Torus φ")
155
  torus_Y = gr.Slider(minimum=-pi, maximum=pi, value=0, label="Torus θ")
156
+ torus_Z = gr.Slider(minimum=-pi, maximum=pi, value=pi / 2, label="Torus ψ")
 
 
157
 
158
  render_btn = gr.Button("Render")
159
  with gr.Column(scale=3):