Hugging Face's logo

Spaces:
elapt1c
/
HROM-V1-TEST
Sleeping

App Files Community
TimurHromek commited on
Commit
77b6e8a
·
1 Parent(s): bfe7166

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ tokenizer_file = hf_hub_download(repo_id=model_repo, filename="tokenizer/hrom_to
22
  tokenizer = Tokenizer.from_file(tokenizer_file)
23
 
24
  # 3. Load model checkpoint
25
- checkpoint_file = hf_hub_download(repo_id=model_repo, filename="checkpoints/HROM-V1.5.pt")
26
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
27
 
28
  def load_model():
 
22
  tokenizer = Tokenizer.from_file(tokenizer_file)
23
 
24
  # 3. Load model checkpoint
25
+ checkpoint_file = hf_hub_download(repo_id=model_repo, filename="HROM-V1.5_Trained-Model/HROM-V1.5.pt")
26
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
27
 
28
  def load_model():