chaenykim commited on
Commit
a04445e
1 Parent(s): 67181a9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -7,8 +7,8 @@ license: mit
7
  import torch
8
  from informer_models import InformerConfig, InformerForSequenceClassification
9
 
10
- # Loading the model
11
  model = InformerForSequenceClassification.from_pretrained("BrachioLab/supernova-classification")
 
12
  model.to(device)
13
  model.eval()
14
  y_true = []
 
7
  import torch
8
  from informer_models import InformerConfig, InformerForSequenceClassification
9
 
 
10
  model = InformerForSequenceClassification.from_pretrained("BrachioLab/supernova-classification")
11
+
12
  model.to(device)
13
  model.eval()
14
  y_true = []