alessandro trinca tornidor commited on
Commit
3db919f
·
1 Parent(s): 32cf4a3

feat: fix wrong typing hints that caused exception on app.py start because of gradio

Browse files
Files changed (1) hide show
  1. lisa_on_cuda/utils/app_helpers.py +3 -3
lisa_on_cuda/utils/app_helpers.py CHANGED
@@ -293,7 +293,7 @@ def prepare_model_vision_tower(_model, args_to_parse, torch_dtype, internal_logg
293
 
294
 
295
  def get_inference_model_by_args(
296
- args_to_parse, internal_logger0: logging = None, inference_decorator: Callable = None, device_map="auto", device="cuda"
297
  ):
298
  """Load model and inference function with arguments. Compatible with ZeroGPU (spaces 0.30.2)
299
 
@@ -317,7 +317,7 @@ def get_inference_model_by_args(
317
  def inference(
318
  input_str: str,
319
  input_image: str | Path | np.ndarray,
320
- internal_logger: logging = None,
321
  embedding_key: str = None
322
  ):
323
  if internal_logger is None:
@@ -471,7 +471,7 @@ def get_gradio_interface(
471
  description=constants.description,
472
  article=article_and_demo_parameters,
473
  examples=constants.examples,
474
- allow_flagging="auto"
475
  )
476
 
477
 
 
293
 
294
 
295
  def get_inference_model_by_args(
296
+ args_to_parse, internal_logger0: logging.Logger = None, inference_decorator: Callable = None, device_map="auto", device="cuda"
297
  ):
298
  """Load model and inference function with arguments. Compatible with ZeroGPU (spaces 0.30.2)
299
 
 
317
  def inference(
318
  input_str: str,
319
  input_image: str | Path | np.ndarray,
320
+ internal_logger: logging.Logger = None,
321
  embedding_key: str = None
322
  ):
323
  if internal_logger is None:
 
471
  description=constants.description,
472
  article=article_and_demo_parameters,
473
  examples=constants.examples,
474
+ # flagging_mode="auto"
475
  )
476
 
477