nalin0503 commited on
Commit
d3f4d08
·
1 Parent(s): c811fda

remove slab toggle for HF

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -283,8 +283,8 @@ def main():
283
  st.subheader("3. Generate Morphing Video")
284
  st.markdown("Once satisfied with your inputs, click below to start the process.")
285
 
286
- # New checkbox for SLAB execution toggle
287
- using_slab = st.checkbox("Using SLAB GPU Cluster?", value=False, help="If enabled, the pipeline command will be prefixed with SLAB cluster execution parameters.")
288
 
289
  if st.button("Run Morphing Pipeline", key="run_pipeline"):
290
  if not (uploaded_image_A and uploaded_image_B):
@@ -341,12 +341,12 @@ def main():
341
  cmd.extend(["--film_fps", str(film_fps)])
342
  cmd.extend(["--film_num_recursions", str(film_recursions)])
343
 
344
- if using_slab:
345
- slab_prefix = [
346
- "srun", "-p", "rtx3090_slab", "-w", "slabgpu05", "--gres=gpu:1",
347
- "--job-name=test", "--kill-on-bad-exit=1"
348
- ]
349
- cmd = slab_prefix + cmd
350
 
351
  st.info("Initializing pipeline. This may take a few minutes...")
352
  progress_bar = st.progress(0)
 
283
  st.subheader("3. Generate Morphing Video")
284
  st.markdown("Once satisfied with your inputs, click below to start the process.")
285
 
286
+ # # New checkbox for SLAB execution toggle
287
+ # using_slab = st.checkbox("Using SLAB GPU Cluster?", value=False, help="If enabled, the pipeline command will be prefixed with SLAB cluster execution parameters.")
288
 
289
  if st.button("Run Morphing Pipeline", key="run_pipeline"):
290
  if not (uploaded_image_A and uploaded_image_B):
 
341
  cmd.extend(["--film_fps", str(film_fps)])
342
  cmd.extend(["--film_num_recursions", str(film_recursions)])
343
 
344
+ # if using_slab:
345
+ # slab_prefix = [
346
+ # "srun", "-p", "rtx3090_slab", "-w", "slabgpu05", "--gres=gpu:1",
347
+ # "--job-name=test", "--kill-on-bad-exit=1"
348
+ # ]
349
+ # cmd = slab_prefix + cmd
350
 
351
  st.info("Initializing pipeline. This may take a few minutes...")
352
  progress_bar = st.progress(0)