fixed runtime error

ChitChat/common/utils.py

@@ -13,8 +13,8 @@ small_model = AutoModelForCausalLM.from_pretrained(small_model_name)
 small_tokenizer = AutoTokenizer.from_pretrained(default_model)
 small_tokenizer.pad_token = small_tokenizer.eos_token
 
-large_model = T5ForConditionalGeneration.from_pretrained(large_model_name)
-large_tokenizer = T5Tokenizer.from_pretrained(large_model_name)
+# large_model = T5ForConditionalGeneration.from_pretrained(large_model_name)
+# large_tokenizer = T5Tokenizer.from_pretrained(large_model_name)
 
 def getChatHistory(user):
     if user.history is None:
@@ -57,6 +57,7 @@ def conversation(user, userInput):
     return small_tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens = True)
 
 def complexChat(userInput):
+    return "hi"
     input_ids = large_tokenizer(userInput, return_tensors="pt").input_ids
     outputs = large_model.generate(input_ids,
                          min_length = 10,

ChitChat/models.py

@@ -1,14 +1,9 @@
 from ChitChat import db 
 from datetime import datetime 
 import uuid
-from flask_login import UserMixin
-from ChitChat import login_manager
 
-@login_manager.user_loader
-def load_user(user_id):
-    return User.query.get(int(user_id)) 
 
-class User(db.Model, UserMixin):
+class User(db.Model):
     id = db.Column(db.Integer, primary_key = True)
     username  = db.Column(db.String(10), unique = True, nullable = False)
     email = db.Column(db.String(100), nullable = False)

requirements.txt

@@ -9,7 +9,6 @@ filelock==3.12.2
 Flask==2.3.2
 Flask-Bcrypt==1.0.1
 Flask-Cors==3.0.10
-Flask-Login==0.6.2
 Flask-RESTful==0.3.10
 Flask-SQLAlchemy==3.0.3
 fsspec==2023.6.0