Update app.py
Browse files
app.py
CHANGED
@@ -1,7 +1,6 @@
|
|
1 |
import os
|
2 |
import json
|
3 |
import pickle
|
4 |
-
import random
|
5 |
import tempfile
|
6 |
import asyncio
|
7 |
import io
|
@@ -18,9 +17,7 @@ from fastapi import FastAPI
|
|
18 |
from fastapi.responses import HTMLResponse
|
19 |
from pydantic import BaseModel
|
20 |
from dotenv import load_dotenv
|
21 |
-
from datetime import datetime
|
22 |
from faker import Faker
|
23 |
-
from random_word import RandomWords
|
24 |
import logging
|
25 |
|
26 |
logging.basicConfig(level=logging.INFO)
|
@@ -99,18 +96,13 @@ async def train_and_save_model():
|
|
99 |
classes.append(intent['tag'])
|
100 |
|
101 |
fake = Faker()
|
102 |
-
|
103 |
-
|
104 |
-
|
105 |
-
|
106 |
-
|
107 |
-
|
108 |
-
|
109 |
-
if random_tag not in classes:
|
110 |
-
classes.append(random_tag)
|
111 |
-
except Exception as e:
|
112 |
-
logger.error("API key either expired or not working. Error: {}".format(str(e)))
|
113 |
-
continue
|
114 |
|
115 |
words = sorted(set(words))
|
116 |
classes = sorted(set(classes))
|
|
|
1 |
import os
|
2 |
import json
|
3 |
import pickle
|
|
|
4 |
import tempfile
|
5 |
import asyncio
|
6 |
import io
|
|
|
17 |
from fastapi.responses import HTMLResponse
|
18 |
from pydantic import BaseModel
|
19 |
from dotenv import load_dotenv
|
|
|
20 |
from faker import Faker
|
|
|
21 |
import logging
|
22 |
|
23 |
logging.basicConfig(level=logging.INFO)
|
|
|
96 |
classes.append(intent['tag'])
|
97 |
|
98 |
fake = Faker()
|
99 |
+
for _ in range(10):
|
100 |
+
random_pattern = f"{fake.sentence()}"
|
101 |
+
random_tag = fake.word()
|
102 |
+
documents.append((nltk.word_tokenize(random_pattern), random_tag))
|
103 |
+
words.extend(nltk.word_tokenize(random_pattern))
|
104 |
+
if random_tag not in classes:
|
105 |
+
classes.append(random_tag)
|
|
|
|
|
|
|
|
|
|
|
106 |
|
107 |
words = sorted(set(words))
|
108 |
classes = sorted(set(classes))
|