Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -8,6 +8,7 @@ import gradio as gr
|
|
8 |
# Load the model and tokenizer
|
9 |
model_path = "ByteDance/Sa2VA-4B"
|
10 |
|
|
|
11 |
from transformers.dynamic_module_utils import get_imports
|
12 |
|
13 |
def fixed_get_imports(filename: str | os.PathLike) -> list[str]:
|
|
|
8 |
# Load the model and tokenizer
|
9 |
model_path = "ByteDance/Sa2VA-4B"
|
10 |
|
11 |
+
from unittest.mock import patch
|
12 |
from transformers.dynamic_module_utils import get_imports
|
13 |
|
14 |
def fixed_get_imports(filename: str | os.PathLike) -> list[str]:
|