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):
|