Werli commited on
Commit
cb15c85
·
verified ·
1 Parent(s): e0a692d

Update modules/florence2.py

Browse files
Files changed (1) hide show
  1. 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)