Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 =
|
| 125 |
-
bpy.context.scene.render.resolution_x =
|
| 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 |
-
|
| 132 |
-
|
| 133 |
|
| 134 |
-
|
| 135 |
-
|
| 136 |
-
|
| 137 |
-
|
| 138 |
-
|
| 139 |
-
|
| 140 |
-
|
| 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):
|