Update app.py

app.py

@@ -76,8 +76,8 @@ st.markdown("""
 # Load the model outside the handle_input() function
 with open('model_saved.pkl', 'rb') as f:
     model = pickle.load(f)
-    if not hasattr(model, 'predict'):
-        st.error("The loaded model does not have a predict method.")
+    if not isinstance(model, str):
+        st.error("The loaded model is not valid.")
 
 def write_top_bar():
     col1, col2, col3 = st.columns([1,10,2])
@@ -110,11 +110,8 @@ def handle_input():
     if len(chat_history) == MAX_HISTORY_LENGTH:
         chat_history = chat_history[:-1]
 
-    def generate_response(question: str) -> str:
-        return model.predict([question])[0]
-
     prompt = input
-    answer = generate_response(prompt)
+    answer = model  # Replace the predict() method with the model itself
 
     chat_history.append((input, answer))