CogwiseAI commited on
Commit
615c973
·
1 Parent(s): 081c79a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -103,9 +103,12 @@ def handle_input():
103
  chat_history = st.session_state["chat_history"]
104
  if len(chat_history) == MAX_HISTORY_LENGTH:
105
  chat_history = chat_history[:-1]
106
-
107
- with open('model_saved.pkl', 'rb') as f:
108
- model = pickle.load(f)
 
 
 
109
 
110
  def generate_response(question: str) -> str:
111
  return model.predict([question])[0]
 
103
  chat_history = st.session_state["chat_history"]
104
  if len(chat_history) == MAX_HISTORY_LENGTH:
105
  chat_history = chat_history[:-1]
106
+
107
+ with open('model_saved.pkl', 'rb') as f:
108
+ model = pickle.load(f)
109
+ if not hasattr(model, 'predict'):
110
+ st.error("The loaded model does not have a predict method.")
111
+ return
112
 
113
  def generate_response(question: str) -> str:
114
  return model.predict([question])[0]