Update services/nlp.py
Browse files- services/nlp.py +1 -4
services/nlp.py
CHANGED
|
@@ -4,7 +4,6 @@ from models.input import Input
|
|
| 4 |
import re
|
| 5 |
from transformers import pipeline
|
| 6 |
from services.utils import clean_text
|
| 7 |
-
import speech_recognition as sr
|
| 8 |
|
| 9 |
intent_recognizer = pipeline(
|
| 10 |
"zero-shot-classification", model="facebook/bart-large-mnli")
|
|
@@ -75,6 +74,4 @@ def generate_response(input: Input, context_from_elasticsearch):
|
|
| 75 |
messages=prompt
|
| 76 |
)
|
| 77 |
|
| 78 |
-
return response.choices[0].message.content
|
| 79 |
-
|
| 80 |
-
|
|
|
|
| 4 |
import re
|
| 5 |
from transformers import pipeline
|
| 6 |
from services.utils import clean_text
|
|
|
|
| 7 |
|
| 8 |
intent_recognizer = pipeline(
|
| 9 |
"zero-shot-classification", model="facebook/bart-large-mnli")
|
|
|
|
| 74 |
messages=prompt
|
| 75 |
)
|
| 76 |
|
| 77 |
+
return response.choices[0].message.content
|
|
|
|
|
|