chandra10 commited on
Commit
b5c3986
·
1 Parent(s): 58ed862

Update Space

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -43,11 +43,13 @@ model.load_state_dict(model_state)
43
  model.eval()
44
 
45
  # Fungsi untuk tokenisasi, stemming, dan bag-of-words
 
 
46
  def tokenize(sentence):
47
  return nltk.word_tokenize(sentence)
48
 
49
  def stem(word):
50
- return PorterStemmer().stem(word.lower())
51
 
52
  def bag_of_words(tokenized_sentence, all_words):
53
  tokenized_sentence = [stem(w) for w in tokenized_sentence]
@@ -55,6 +57,7 @@ def bag_of_words(tokenized_sentence, all_words):
55
  for idx, w in enumerate(all_words):
56
  if w in tokenized_sentence:
57
  bag[idx] = 1.0
 
58
  return bag
59
 
60
  # Fungsi prediksi
 
43
  model.eval()
44
 
45
  # Fungsi untuk tokenisasi, stemming, dan bag-of-words
46
+ stemmer = PorterStemmer()
47
+
48
  def tokenize(sentence):
49
  return nltk.word_tokenize(sentence)
50
 
51
  def stem(word):
52
+ return stemmer.stem(word.lower())
53
 
54
  def bag_of_words(tokenized_sentence, all_words):
55
  tokenized_sentence = [stem(w) for w in tokenized_sentence]
 
57
  for idx, w in enumerate(all_words):
58
  if w in tokenized_sentence:
59
  bag[idx] = 1.0
60
+
61
  return bag
62
 
63
  # Fungsi prediksi