updated requirements

ChitChat/common/utils.py

@@ -1,10 +1,8 @@
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from transformers import T5Tokenizer, T5ForConditionalGeneration 
 import torch
-from flask import current_app, session
+from flask import current_app
 from ChitChat import db
-import numpy as np
-import pandas as pd
 
 small_model_name = 'Th3BossC/DialoGPT-medium-AICLUB_NITC'
 default_model = 'microsoft/DialoGPT-medium'
@@ -53,7 +51,6 @@ def conversation(user, userInput):
         top_p = 0.7,
         temperature = 0.8
     )
-    # print(f"chat_history_ids : {type(chat_history_ids)}")
     saveChatHistory(user, chat_history_ids)
     return small_tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens = True)
 

ChitChat/resources/routes.py

@@ -1,9 +1,8 @@
-from flask import Blueprint, request, session
+from flask import Blueprint, request
 from flask_restful import Api, Resource 
 from ChitChat.models import User
 from ChitChat import bcrypt, db
 from ChitChat.common.utils import conversation, complexChat
-import secrets
 resources = Blueprint('resources', __name__)
 api = Api(resources)
 

requirements.txt

@@ -22,6 +22,7 @@ mpmath==1.3.0
 networkx==3.1
 numpy==1.24.3
 packaging==23.1
+python-dateutil==2.8.2
 pytz==2023.3
 PyYAML==6.0
 regex==2023.6.3
@@ -36,5 +37,6 @@ torch==2.0.1
 tqdm==4.65.0
 transformers==4.30.2
 typing_extensions==4.6.3
+tzdata==2023.3
 urllib3==2.0.3
 Werkzeug==2.3.6