AdamF92 commited on
Commit
306fd6d
·
verified ·
1 Parent(s): aa6c0d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from rxlm.training.tokenizer import load_tokenizer_from_hf_hub
9
  HF_TOKEN = os.environ.get("HF_TOKEN")
10
 
11
  tokenizer = load_tokenizer_from_hf_hub('ReactiveAI/RxT-Beta-Micro-Supervised-AI', token=HF_TOKEN)
12
- model = RxTBeta.from_pretrained('RxT-Beta-Micro-Supervised-AI', token=HF_TOKEN, tokenizer=tokenizer)
13
  model.share_components()
14
 
15
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
9
  HF_TOKEN = os.environ.get("HF_TOKEN")
10
 
11
  tokenizer = load_tokenizer_from_hf_hub('ReactiveAI/RxT-Beta-Micro-Supervised-AI', token=HF_TOKEN)
12
+ model = RxTBeta.from_pretrained('ReactiveAI/RxT-Beta-Micro-Supervised-AI', token=HF_TOKEN, tokenizer=tokenizer)
13
  model.share_components()
14
 
15
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')