owaiskha9654 commited on
Commit
b450063
β€’
1 Parent(s): f1b4b0b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -5,6 +5,7 @@ from typing import Dict
5
  from torch.utils.data import TensorDataset, DataLoader, RandomSampler, SequentialSampler
6
  from transformers import BertForSequenceClassification,BertTokenizer
7
 
 
8
  num_labels=14
9
  #Loading Model and Tokenizer from Hugging Face Spaces
10
  model = BertForSequenceClassification.from_pretrained("owaiskha9654/Multi-Label-Classification-of-PubMed-Articles", num_labels=num_labels)
 
5
  from torch.utils.data import TensorDataset, DataLoader, RandomSampler, SequentialSampler
6
  from transformers import BertForSequenceClassification,BertTokenizer
7
 
8
+
9
  num_labels=14
10
  #Loading Model and Tokenizer from Hugging Face Spaces
11
  model = BertForSequenceClassification.from_pretrained("owaiskha9654/Multi-Label-Classification-of-PubMed-Articles", num_labels=num_labels)