mew77 commited on
Commit
a71142d
·
verified ·
1 Parent(s): 8a5d7ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -3,7 +3,7 @@ from PIL import Image
3
  import os
4
  from datetime import datetime
5
  from hf_model import HFModel # Import your HFModel class
6
- from .mew_log import log_info, log_error # Import your custom logging methods
7
 
8
  # Initialize the model
9
  MODEL_NAME = "deepseek-ai/DeepSeek-Coder-V2-Lite-Base"
@@ -14,7 +14,7 @@ def initialize_model():
14
  log_info("Model initialized successfully.")
15
  return model
16
  except Exception as e:
17
- log_error(f"Failed to initialize model: {e}")
18
  st.error(f"Failed to initialize model: {e}")
19
  return None
20
 
@@ -80,7 +80,7 @@ def main():
80
  st.write(f"**{role.capitalize()}:** {content}")
81
 
82
  except Exception as e:
83
- log_error(f"Error during chat: {e}")
84
  st.error(f"An error occurred: {e}")
85
 
86
  # Clear chat history button
 
3
  import os
4
  from datetime import datetime
5
  from hf_model import HFModel # Import your HFModel class
6
+ from mew_log import log_info, log_error # Import your custom logging methods
7
 
8
  # Initialize the model
9
  MODEL_NAME = "deepseek-ai/DeepSeek-Coder-V2-Lite-Base"
 
14
  log_info("Model initialized successfully.")
15
  return model
16
  except Exception as e:
17
+ log_error(f"Failed to initialize model: {e}", e)
18
  st.error(f"Failed to initialize model: {e}")
19
  return None
20
 
 
80
  st.write(f"**{role.capitalize()}:** {content}")
81
 
82
  except Exception as e:
83
+ log_error(f"Error during chat: {e}", e)
84
  st.error(f"An error occurred: {e}")
85
 
86
  # Clear chat history button