Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
7d8772d
1
Parent(s):
6535ded
docs: little update
Browse files- app.py +2 -0
- 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 |
|