Hugging Face's logo

Spaces:
KLeedrug
/
EMO_AI_alpha
Runtime error

App Files Community
KLeedrug commited on
Commit
d638f0b
•
1 Parent(s): 5854256

add the latest weight "arch1_unfreeze_all.pt" to repo

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -56,7 +56,8 @@ class EmoModel(nn.Module):
56
 
57
 
58
  from pathlib import Path
59
- pretrained_path = "on_plurk_new_fix_data_arch_1_epoch_2_bs_16.pt"
 
60
  assert Path(pretrained_path).is_file()
61
 
62
  model = EmoModel(AutoModelWithLMHead.from_pretrained("distilroberta-base").base_model)
 
56
 
57
 
58
  from pathlib import Path
59
+ #pretrained_path = "on_plurk_new_fix_data_arch_1_epoch_2_bs_16.pt"
60
+ pretrained_path = "arch1_unfreeze_all.pt" # the latest weights!
61
  assert Path(pretrained_path).is_file()
62
 
63
  model = EmoModel(AutoModelWithLMHead.from_pretrained("distilroberta-base").base_model)