doublesizebed commited on
Commit
cde87a2
·
1 Parent(s): e65b605

updates html

Browse files
Files changed (2) hide show
  1. app.py +13 -0
  2. 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