yixin1121 commited on
Commit
544e91d
·
verified ·
1 Parent(s): 513e1fb

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +106 -0
app.py CHANGED
@@ -99,7 +99,7 @@ def generate(video, textbox_in, first_run, state, state_):
99
  return (state, state_, state.to_gradio_chatbot(), False, gr.update(value=None, interactive=True), images_tensor, gr.update(value=image1 if os.path.exists(image1) else None, interactive=True), gr.update(value=video if os.path.exists(video) else None, interactive=True))
100
 
101
 
102
- device = "cuda"
103
  handler = Infer(device)
104
  # handler.model.to(dtype=dtype)
105
  if not os.path.exists("temp"):
 
99
  return (state, state_, state.to_gradio_chatbot(), False, gr.update(value=None, interactive=True), images_tensor, gr.update(value=image1 if os.path.exists(image1) else None, interactive=True), gr.update(value=video if os.path.exists(video) else None, interactive=True))
100
 
101
 
102
+ device = "cpu"
103
  handler = Infer(device)
104
  # handler.model.to(dtype=dtype)
105
  if not os.path.exists("temp"):
requirements.txt ADDED
@@ -0,0 +1,106 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ aiofiles==23.2.1
2
+ altair==5.3.0
3
+ annotated-types==0.7.0
4
+ anyio==4.4.0
5
+ attrs==23.2.0
6
+ certifi==2024.7.4
7
+ charset-normalizer==3.3.2
8
+ click==8.1.7
9
+ clip @ git+https://github.com/openai/CLIP.git@dcba3cb2e2827b402d2701e7e1c7d9fed8a20ef1
10
+ cmake==3.30.0
11
+ contourpy==1.1.1
12
+ cycler==0.12.1
13
+ dnspython==2.6.1
14
+ email_validator==2.2.0
15
+ exceptiongroup==1.2.2
16
+ fastapi==0.111.0
17
+ fastapi-cli==0.0.4
18
+ ffmpeg==1.4
19
+ ffmpy==0.3.2
20
+ filelock==3.15.4
21
+ fonttools==4.53.1
22
+ fsspec==2024.6.1
23
+ ftfy==6.2.0
24
+ gradio==3.46.0
25
+ gradio_client==0.5.3
26
+ h11==0.14.0
27
+ httpcore==1.0.5
28
+ httptools==0.6.1
29
+ httpx==0.27.0
30
+ huggingface-hub==0.23.4
31
+ idna==3.7
32
+ importlib_resources==6.4.0
33
+ Jinja2==3.1.4
34
+ joblib==1.4.2
35
+ jsonschema==4.23.0
36
+ jsonschema-specifications==2023.12.1
37
+ kiwisolver==1.4.5
38
+ lit==18.1.8
39
+ markdown-it-py==3.0.0
40
+ MarkupSafe==2.1.5
41
+ matplotlib==3.7.5
42
+ mdurl==0.1.2
43
+ mpmath==1.3.0
44
+ networkx==3.1
45
+ numpy==1.24.4
46
+ nvidia-cublas-cu11==11.10.3.66
47
+ nvidia-cuda-cupti-cu11==11.7.101
48
+ nvidia-cuda-nvrtc-cu11==11.7.99
49
+ nvidia-cuda-runtime-cu11==11.7.99
50
+ nvidia-cudnn-cu11==8.5.0.96
51
+ nvidia-cufft-cu11==10.9.0.58
52
+ nvidia-curand-cu11==10.2.10.91
53
+ nvidia-cusolver-cu11==11.4.0.1
54
+ nvidia-cusparse-cu11==11.7.4.91
55
+ nvidia-nccl-cu11==2.14.3
56
+ nvidia-nvtx-cu11==11.7.91
57
+ opencv-python==4.10.0.84
58
+ orjson==3.10.6
59
+ packaging==24.1
60
+ pandas==2.0.3
61
+ pillow==10.4.0
62
+ pkgutil_resolve_name==1.3.10
63
+ pydantic==2.8.2
64
+ pydantic_core==2.20.1
65
+ pydub==0.25.1
66
+ Pygments==2.18.0
67
+ pyparsing==3.1.2
68
+ python-dateutil==2.9.0.post0
69
+ python-dotenv==1.0.1
70
+ python-multipart==0.0.9
71
+ pytz==2024.1
72
+ PyYAML==6.0.1
73
+ referencing==0.35.1
74
+ regex==2024.5.15
75
+ requests==2.32.3
76
+ rich==13.7.1
77
+ rpds-py==0.19.0
78
+ ruff==0.5.1
79
+ sacremoses==0.1.1
80
+ semantic-version==2.10.0
81
+ sentencepiece==0.2.0
82
+ shellingham==1.5.4
83
+ six==1.16.0
84
+ sniffio==1.3.1
85
+ starlette==0.37.2
86
+ sympy==1.13.0
87
+ tokenizers==0.12.1
88
+ tomlkit==0.12.0
89
+ toolz==0.12.1
90
+ torch==2.0.0
91
+ torchvision==0.15.1
92
+ tqdm==4.66.4
93
+ transformers==4.18.0
94
+ triton==2.0.0
95
+ typer==0.12.3
96
+ typing_extensions==4.12.2
97
+ tzdata==2024.1
98
+ ujson==5.10.0
99
+ urllib3==2.2.2
100
+ uvicorn==0.30.1
101
+ uvloop==0.19.0
102
+ watchfiles==0.22.0
103
+ wcwidth==0.2.13
104
+ websockets==11.0.3
105
+ Werkzeug==3.0.3
106
+ zipp==3.19.2