Update main.py

main.py

@@ -2,10 +2,9 @@ from flask import Flask
 from gradio_client import Client
 import time
 from multiprocessing import Process
-import logging
+from transformers import Trainer
 
-
-logger = logging.getLogger(__name__)
+trainer = Trainer()
 app = Flask(__name__)
 
 @app.route('/')
@@ -23,13 +22,13 @@ def api(query):
 
 def record_loop():
    while True:
-        logger.info('Loop Running')
+        trainer.log('Loop Running')
         print("loop running")
         time.sleep(5)
 
 
 if __name__ == "__main__":
-   logger.info('App Starting...')
+   trainer.log('App Starting...')
    p = Process(target=record_loop)
    p.start()  
    app.run(debug=True, use_reloader=False)