Update app.py
Browse files
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=
|
| 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]
|