HikariDawn commited on
Commit
7d8772d
·
1 Parent(s): 6535ded

docs: little update

Browse files
Files changed (2) hide show
  1. app.py +2 -0
  2. requirements.txt +1 -4
app.py CHANGED
@@ -10,6 +10,7 @@ import time
10
  import imageio
11
  import random
12
  import ast
 
13
  import gradio as gr
14
  from omegaconf import OmegaConf
15
  from PIL import Image
@@ -511,6 +512,7 @@ def sample_traj_by_length(points, num_samples):
511
 
512
 
513
 
 
514
  def inference(inference_canvas, visual_canvas, text_prompt, traj_lists, main_reference_img,
515
  resized_height, resized_width, top_left_height, top_left_width, bottom_right_height, bottom_right_width):
516
 
 
10
  import imageio
11
  import random
12
  import ast
13
+ import spaces
14
  import gradio as gr
15
  from omegaconf import OmegaConf
16
  from PIL import Image
 
512
 
513
 
514
 
515
+ @spaces.GPU(duration=120)
516
  def inference(inference_canvas, visual_canvas, text_prompt, traj_lists, main_reference_img,
517
  resized_height, resized_width, top_left_height, top_left_width, bottom_right_height, bottom_right_width):
518
 
requirements.txt CHANGED
@@ -4,21 +4,18 @@ opencv-python
4
  pyiqa
5
  numpy==1.26.0
6
  ffmpeg-python
7
- bitsandbytes
8
  pyarrow
9
  omegaconf
10
  peft>=0.15.0
11
  transformers>=4.56.2 # Install in the newest version
12
  git+https://github.com/huggingface/diffusers.git
13
  sentencepiece
14
- qwen-vl-utils[decord]==0.0.8
15
  scikit-learn
16
  matplotlib
17
  gradio
18
  imageio-ffmpeg
19
- bitsandbytes
20
  git+https://github.com/facebookresearch/segment-anything.git
21
- git+https://github.com/facebookresearch/sam2.git
22
  accelerate
23
  hf-transfer
24
 
 
4
  pyiqa
5
  numpy==1.26.0
6
  ffmpeg-python
 
7
  pyarrow
8
  omegaconf
9
  peft>=0.15.0
10
  transformers>=4.56.2 # Install in the newest version
11
  git+https://github.com/huggingface/diffusers.git
12
  sentencepiece
13
+ # qwen-vl-utils[decord]==0.0.8
14
  scikit-learn
15
  matplotlib
16
  gradio
17
  imageio-ffmpeg
 
18
  git+https://github.com/facebookresearch/segment-anything.git
 
19
  accelerate
20
  hf-transfer
21