Thouph commited on
Commit
58d222b
·
verified ·
1 Parent(s): 18a9701

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -3,10 +3,10 @@ torch.set_grad_enabled(False)
3
  import gradio as gr
4
  from datetime import datetime
5
  from inference_tagger_standalone import *
6
- from huggingface_hub import hf_hub_download
7
- hf_hub_download(repo_id="lodestones/tagger-experiment", filename="tagger_proto.safetensors", local_dir=".")
8
- # import os
9
- # os.system("wget https://huggingface.co/lodestones/tagger-experiment/resolve/main/tagger_proto.safetensors")
10
  model = Tagger(checkpoint_path="./tagger_proto.safetensers", vocab_path="./tagger_vocab.json", max_size=1024)
11
  model.eval()
12
 
 
3
  import gradio as gr
4
  from datetime import datetime
5
  from inference_tagger_standalone import *
6
+ # from huggingface_hub import hf_hub_download
7
+ # hf_hub_download(repo_id="lodestones/tagger-experiment", filename="tagger_proto.safetensors", local_dir=".")
8
+ import os
9
+ os.system("wget https://huggingface.co/lodestones/tagger-experiment/resolve/main/tagger_proto.safetensors")
10
  model = Tagger(checkpoint_path="./tagger_proto.safetensers", vocab_path="./tagger_vocab.json", max_size=1024)
11
  model.eval()
12