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 |
|