vsrinivas commited on
Commit
82a135d
1 Parent(s): 76c73ff

Update funcs.py

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