Update funcs.py

funcs.py

@@ -10,7 +10,7 @@ from PIL import Image
 import os
 from datetime import datetime
 from openai import OpenAI
-from ai71 import AI71
+# from ai71 import AI71
 
 if torch.cuda.is_available():
     model = model.to('cuda')
@@ -22,8 +22,8 @@ with open ('emotion_group_labels.txt') as file:
 embed_model = SentenceTransformer('paraphrase-MiniLM-L6-v2')
 classifier = pipeline("zero-shot-classification", model ='facebook/bart-large-mnli')
 
-AI71_API_KEY = os.getenv('AI71_API_KEY')
-print(AI71_API_KEY)
+# AI71_API_KEY = os.getenv('AI71_API_KEY')
+# print(AI71_API_KEY)
 huggingface_token = os.getenv('hf_token')
 print("hf-token:",huggingface_token)
 
@@ -96,7 +96,6 @@ def get_doc_response_emotions(user_message, therapy_session_conversation):
     print(f"therapy_session_conversation: {therapy_session_conversation}")
     return '', therapy_session_conversation, emotions_msg
 
-from datetime import datetime
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch