Spaces:
Running
on
Zero
Running
on
Zero
Update app.py (#5)
Browse files- Update app.py (6af2826e4ddc47ff19a6397ab87a674056419ced)
Co-authored-by: YuqianYuan <[email protected]>
app.py
CHANGED
@@ -1,7 +1,7 @@
|
|
|
|
1 |
import gradio as gr
|
2 |
import numpy as np
|
3 |
import os
|
4 |
-
#os.system("pip install torch==2.5.1 --force-reinstall --extra-index-url https://download.pytorch.org/whl/cu124")
|
5 |
import torch
|
6 |
from transformers import SamModel, SamProcessor
|
7 |
from PIL import Image
|
@@ -13,7 +13,6 @@ sys.path.append('./')
|
|
13 |
from videollama3 import disable_torch_init, model_init, mm_infer, get_model_output
|
14 |
from videollama3.mm_utils import load_images
|
15 |
from videollama3.mm_utils import load_video
|
16 |
-
import spaces
|
17 |
|
18 |
|
19 |
color_rgb = (1.0, 1.0, 1.0)
|
|
|
1 |
+
import spaces
|
2 |
import gradio as gr
|
3 |
import numpy as np
|
4 |
import os
|
|
|
5 |
import torch
|
6 |
from transformers import SamModel, SamProcessor
|
7 |
from PIL import Image
|
|
|
13 |
from videollama3 import disable_torch_init, model_init, mm_infer, get_model_output
|
14 |
from videollama3.mm_utils import load_images
|
15 |
from videollama3.mm_utils import load_video
|
|
|
16 |
|
17 |
|
18 |
color_rgb = (1.0, 1.0, 1.0)
|