Tonic commited on
Commit
621c8c9
•
1 Parent(s): 17d4493
Files changed (3) hide show
  1. .gitignore +1 -0
  2. app.py +7 -0
  3. requirements.txt +2 -1
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ llava_env/
app.py CHANGED
@@ -1,4 +1,11 @@
1
  import gradio as gr
 
 
 
 
 
 
 
2
  import torch
3
  from llava.model.builder import load_pretrained_model
4
  from llava.mm_utils import get_model_name_from_path, process_images, tokenizer_image_token
 
1
  import gradio as gr
2
+ import subprocess # 🥲
3
+
4
+ subprocess.run(
5
+ "pip install git+https://github.com/LLaVA-VL/LLaVA-NeXT.git",
6
+ shell=True,
7
+ )
8
+
9
  import torch
10
  from llava.model.builder import load_pretrained_model
11
  from llava.mm_utils import get_model_name_from_path, process_images, tokenizer_image_token
requirements.txt CHANGED
@@ -3,4 +3,5 @@ numpy
3
  transformers
4
  torch
5
  torchvision
6
- git+https://github.com/LLaVA-VL/LLaVA-NeXT.git
 
 
3
  transformers
4
  torch
5
  torchvision
6
+ decord
7
+ einops