Spaces:
Paused
Paused
Commit
·
cde87a2
1
Parent(s):
e65b605
updates html
Browse files- app.py +13 -0
- requirements.txt +2 -1
app.py
CHANGED
|
@@ -8,6 +8,7 @@ from flask_cors import CORS
|
|
| 8 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 9 |
from deep_translator import GoogleTranslator
|
| 10 |
from textblob import TextBlob
|
|
|
|
| 11 |
from parler_tts import ParlerTTSForConditionalGeneration
|
| 12 |
|
| 13 |
# Flask setup
|
|
@@ -19,6 +20,18 @@ CORS(app)
|
|
| 19 |
AUDIO_FOLDER = '/static/audio'
|
| 20 |
os.makedirs(AUDIO_FOLDER, exist_ok=True)
|
| 21 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 22 |
|
| 23 |
class ChatBot:
|
| 24 |
def __init__(self):
|
|
|
|
| 8 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 9 |
from deep_translator import GoogleTranslator
|
| 10 |
from textblob import TextBlob
|
| 11 |
+
import nltk
|
| 12 |
from parler_tts import ParlerTTSForConditionalGeneration
|
| 13 |
|
| 14 |
# Flask setup
|
|
|
|
| 20 |
AUDIO_FOLDER = '/static/audio'
|
| 21 |
os.makedirs(AUDIO_FOLDER, exist_ok=True)
|
| 22 |
|
| 23 |
+
try:
|
| 24 |
+
nltk.data.find('corpora/brown')
|
| 25 |
+
except LookupError:
|
| 26 |
+
nltk.download('brown')
|
| 27 |
+
|
| 28 |
+
try:
|
| 29 |
+
nltk.data.find('tokenizers/punkt')
|
| 30 |
+
nltk.data.find('tokenizers/punkt_tab')
|
| 31 |
+
except LookupError:
|
| 32 |
+
nltk.download('punkt')
|
| 33 |
+
nltk.download('punkt_tab')
|
| 34 |
+
|
| 35 |
|
| 36 |
class ChatBot:
|
| 37 |
def __init__(self):
|
requirements.txt
CHANGED
|
@@ -10,4 +10,5 @@ flask-cors
|
|
| 10 |
soundfile
|
| 11 |
textblob
|
| 12 |
malaya
|
| 13 |
-
parler_tts
|
|
|
|
|
|
| 10 |
soundfile
|
| 11 |
textblob
|
| 12 |
malaya
|
| 13 |
+
parler_tts
|
| 14 |
+
nltk
|