mrcuddle commited on
Commit
0679594
·
verified ·
1 Parent(s): 32d631e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ import spaces
8
  pipeline = I2VGenXLPipeline.from_pretrained("ali-vilab/i2vgen-xl", torch_dtype=torch.float16, variant="fp16")
9
  pipeline.enable_model_cpu_offload()
10
 
11
- @spaces.GPU
12
  def generate_gif(image, prompt, negative_prompt, num_inference_steps, guidance_scale, seed):
13
  # Load the image
14
  image = load_image(image).convert("RGB")
@@ -39,7 +39,7 @@ iface = gr.Interface(
39
  gr.Image(type="filepath", label="Input Image"),
40
  gr.Textbox(lines=2, placeholder="Enter your prompt here...", label="Prompt"),
41
  gr.Textbox(lines=2, placeholder="Enter your negative prompt here...", label="Negative Prompt"),
42
- gr.Slider(1, 100, step=1, value=50, label="Number of Inference Steps"),
43
  gr.Slider(1, 20, step=0.1, value=9.0, label="Guidance Scale"),
44
  gr.Number(label="Seed", value=8888)
45
  ],
 
8
  pipeline = I2VGenXLPipeline.from_pretrained("ali-vilab/i2vgen-xl", torch_dtype=torch.float16, variant="fp16")
9
  pipeline.enable_model_cpu_offload()
10
 
11
+ @spaces.GPU(duration=240)
12
  def generate_gif(image, prompt, negative_prompt, num_inference_steps, guidance_scale, seed):
13
  # Load the image
14
  image = load_image(image).convert("RGB")
 
39
  gr.Image(type="filepath", label="Input Image"),
40
  gr.Textbox(lines=2, placeholder="Enter your prompt here...", label="Prompt"),
41
  gr.Textbox(lines=2, placeholder="Enter your negative prompt here...", label="Negative Prompt"),
42
+ gr.Slider(1, 50, step=1, value=15, label="Number of Inference Steps"),
43
  gr.Slider(1, 20, step=0.1, value=9.0, label="Guidance Scale"),
44
  gr.Number(label="Seed", value=8888)
45
  ],