Update app.py

app.py

@@ -8,8 +8,6 @@ os.system('pip install ./gaussiansplatting/submodules/diff-gaussian-rasterizatio
 os.system('pip install ./gaussiansplatting/submodules/simple-knn')
 os.system('pip install git+https://github.com/KAIR-BAIR/[email protected]')
 os.system('pip install git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch')
-os.system('git clone https://huggingface.co/datasets/tiange/Cap3D')
-# os.system('git clone https://huggingface.co/datasets/tiange/Cap3D')
 
 example_inputs = [[
         "A fox."
@@ -32,14 +30,15 @@ example_outputs_1 = [
 
 
 
-def main(prompt, CFG, seed):
+def main(prompt, iteration,CFG, seed):
     if [prompt] in example_inputs:
         return example_outputs_1[example_inputs.index([prompt])]
     seed = int(seed)
+    iteration = int(iteration)
     print('==> User Prompt:', prompt)
     timestamp = datetime.now().strftime("@%Y%m%d-%H%M%S")
     subprocess.run([
-                    f'python launch.py --config configs/gaussiandreamer-sd.yaml --train --gpu 0 system.prompt_processor.prompt="{prompt}" seed={seed} system.guidance.guidance_scale={CFG} use_timestamp=False timestamp="{timestamp}" '],
+                    f'python launch.py --config configs/gaussiandreamer-sd.yaml --train --gpu 0 system.prompt_processor.prompt="{prompt}" seed={seed} system.guidance.guidance_scale={CFG} trainer.max_steps={iteration} use_timestamp=False timestamp="{timestamp}" '],
                 shell=True)
     path= os.path.join("./outputs/gaussiandreamer-sd",f'{prompt.replace(" ","_")}{timestamp}',"save/it1200-test.mp4")
     print('==> Save path:', path)
@@ -52,6 +51,7 @@ with gr.Blocks() as demo:
                 Note that this demo is running on A10G, the running time might be longer than the reported 35 minutes (5000 iterations) on A100.<br> \
                 &copy; This Gradio space was developed by Haodong LI.")
     gr.Interface(fn=main, inputs=[gr.Textbox(lines=2, value="A portrait of IRONMAN, white hair, head, photorealistic, 8K, HDR.", label="Your prompt"),
+            gr.Slider(0, 2000, value=1200, label="Number of iteration"),
             gr.Slider(80, 200, value=100, label="CFG"),
             gr.Number(value=0, label="Seed")], 
         outputs=["playable_video"],