Spaces:
Running
Running
Update modules/florence2.py
Browse files- modules/florence2.py +1 -0
modules/florence2.py
CHANGED
@@ -3,6 +3,7 @@ from PIL import Image,ImageDraw,ImageFont
|
|
3 |
from unittest.mock import patch
|
4 |
from transformers.dynamic_module_utils import get_imports
|
5 |
from transformers import AutoProcessor,AutoModelForCausalLM
|
|
|
6 |
|
7 |
def fixed_get_imports(filename:str|os.PathLike)->list[str]:
|
8 |
if not str(filename).endswith('/modeling_florence2.py'):return get_imports(filename)
|
|
|
3 |
from unittest.mock import patch
|
4 |
from transformers.dynamic_module_utils import get_imports
|
5 |
from transformers import AutoProcessor,AutoModelForCausalLM
|
6 |
+
import gradio as gr
|
7 |
|
8 |
def fixed_get_imports(filename:str|os.PathLike)->list[str]:
|
9 |
if not str(filename).endswith('/modeling_florence2.py'):return get_imports(filename)
|