Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
-import time  # Import the time module
+import time  
 from model import load_model, generate_image
-from logger import log
+from logger import log_message
 
 def main():
     # Title of the app