Alexvatti commited on
Commit
5bf31ba
·
verified ·
1 Parent(s): 868fbbe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,7 +57,7 @@ print("✅ Model and vocabulary loaded successfully!")
57
  def simple_tokenize(text):
58
  return re.findall(r"\b\w+\b", text.lower())
59
 
60
- def predict(text, model, vocab, max_len=1000):
61
  model.eval()
62
  tokens = simple_tokenize(text.lower())
63
  token_ids = [vocab.get(word, vocab['<UNK>']) for word in tokens]
 
57
  def simple_tokenize(text):
58
  return re.findall(r"\b\w+\b", text.lower())
59
 
60
+ def predict(text, model, vocab, max_len=100):
61
  model.eval()
62
  tokens = simple_tokenize(text.lower())
63
  token_ids = [vocab.get(word, vocab['<UNK>']) for word in tokens]