Update funcs.py
Browse files
funcs.py
CHANGED
@@ -11,6 +11,7 @@ import os
|
|
11 |
from datetime import datetime
|
12 |
from openai import OpenAI
|
13 |
from ai71 import AI71
|
|
|
14 |
|
15 |
if torch.cuda.is_available():
|
16 |
model = model.to('cuda')
|
@@ -159,7 +160,7 @@ def summarize_and_recommend(therapy_session_conversation):
|
|
159 |
full_recommendations = full_recommendations.replace('User:', '').strip()
|
160 |
print("\n")
|
161 |
print("Full recommendations:", full_recommendations)
|
162 |
-
chatbox=[]
|
163 |
return full_summary, full_recommendations, chatbox
|
164 |
|
165 |
|
|
|
11 |
from datetime import datetime
|
12 |
from openai import OpenAI
|
13 |
from ai71 import AI71
|
14 |
+
import gradio as gr
|
15 |
|
16 |
if torch.cuda.is_available():
|
17 |
model = model.to('cuda')
|
|
|
160 |
full_recommendations = full_recommendations.replace('User:', '').strip()
|
161 |
print("\n")
|
162 |
print("Full recommendations:", full_recommendations)
|
163 |
+
chatbox=gr.Chatbot(label="Therapy Session Conversation",value =[[None, 'Therapist: Hello, What can I do for you?']], height=300)
|
164 |
return full_summary, full_recommendations, chatbox
|
165 |
|
166 |
|