atanu0491 commited on
Commit
db2835b
·
1 Parent(s): 6a66e89

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -5,7 +5,7 @@ import warnings
5
  warnings.filterwarnings('ignore')
6
  import re, flair, random, time
7
  from bnlp import BasicTokenizer
8
- from flair.data import Corpus
9
  from flair.datasets import ColumnCorpus
10
  #from flair.embeddings import TransformerWordEmbeddings
11
  from flair.models import SequenceTagger
@@ -37,8 +37,9 @@ if choice == 'Text Input':
37
  # columns = {0: 'text'}
38
  # corpus: Corpus = ColumnCorpus('bengali_pos/', columns, train_file='train.txt', test_file='test.txt', dev_file='valid.txt')
39
  # '''
 
40
  model = load_model('best-model-002.pt')
41
- print(model.predict(input_data))
42
  #print(input_data.to_tagged_string())
43
  #result_mean = model.evaluate(corpus.test, gold_label_type='pos',mini_batch_size=32, out_path=f"pred_pos.txt")
44
 
 
5
  warnings.filterwarnings('ignore')
6
  import re, flair, random, time
7
  from bnlp import BasicTokenizer
8
+ from flair.data import Corpus, Sentence
9
  from flair.datasets import ColumnCorpus
10
  #from flair.embeddings import TransformerWordEmbeddings
11
  from flair.models import SequenceTagger
 
37
  # columns = {0: 'text'}
38
  # corpus: Corpus = ColumnCorpus('bengali_pos/', columns, train_file='train.txt', test_file='test.txt', dev_file='valid.txt')
39
  # '''
40
+ sentence = Sentence(input_data)
41
  model = load_model('best-model-002.pt')
42
+ print(model.predict(sentence))
43
  #print(input_data.to_tagged_string())
44
  #result_mean = model.evaluate(corpus.test, gold_label_type='pos',mini_batch_size=32, out_path=f"pred_pos.txt")
45