Update funcs.py
Browse files
funcs.py
CHANGED
@@ -25,8 +25,6 @@ classifier = pipeline("zero-shot-classification", model ='facebook/bart-large-mn
|
|
25 |
AI71_BASE_URL = "https://api.ai71.ai/v1/"
|
26 |
AI71_API_KEY = os.getenv('AI71_API_KEY')
|
27 |
|
28 |
-
session_conversation=[]
|
29 |
-
|
30 |
# Detect emotions from patient dialogues
|
31 |
def detect_emotions(text):
|
32 |
emotion = classifier(text, candidate_labels=emotion_group_labels, batch_size=16)
|
@@ -157,5 +155,5 @@ class process_session():
|
|
157 |
full_recommendations = full_recommendations.replace('User:', '').strip()
|
158 |
print("\n")
|
159 |
print("Full recommendations:", full_recommendations)
|
160 |
-
session_conversation=[]
|
161 |
return full_summary, full_recommendations
|
|
|
25 |
AI71_BASE_URL = "https://api.ai71.ai/v1/"
|
26 |
AI71_API_KEY = os.getenv('AI71_API_KEY')
|
27 |
|
|
|
|
|
28 |
# Detect emotions from patient dialogues
|
29 |
def detect_emotions(text):
|
30 |
emotion = classifier(text, candidate_labels=emotion_group_labels, batch_size=16)
|
|
|
155 |
full_recommendations = full_recommendations.replace('User:', '').strip()
|
156 |
print("\n")
|
157 |
print("Full recommendations:", full_recommendations)
|
158 |
+
self.session_conversation=[]
|
159 |
return full_summary, full_recommendations
|