laurenssam commited on
Commit
2f3977a
·
1 Parent(s): 64533c0
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ from threading import Thread
7
  import logging
8
  import gradio as gr
9
  import torch
10
- from huggingface_hub import snapshot_download
11
 
12
 
13
  from tinyllava.model.builder import load_pretrained_model
@@ -446,7 +446,7 @@ if __name__ == "__main__":
446
  logger = logging.getLogger(__name__)
447
  logger.info(gr.__version__)
448
  args = parse_args()
449
- model_name = snapshot_download(repo_id="laurenssam/privacy-aware-visual-language-models", filename="tinyllava_demo")
450
  print(model_name)
451
  tokenizer, model, image_processor, context_len = load_pretrained_model(
452
  model_path=model_name,
 
7
  import logging
8
  import gradio as gr
9
  import torch
10
+ from huggingface_hub import hf_hub_download
11
 
12
 
13
  from tinyllava.model.builder import load_pretrained_model
 
446
  logger = logging.getLogger(__name__)
447
  logger.info(gr.__version__)
448
  args = parse_args()
449
+ model_name = hf_hub_download(repo_id="laurenssam/privacy-aware-visual-language-models", filename="tinyllava_demo")
450
  print(model_name)
451
  tokenizer, model, image_processor, context_len = load_pretrained_model(
452
  model_path=model_name,