Update my_model/utilities/ui_manager.py

my_model/utilities/ui_manager.py

@@ -1,6 +1,6 @@
 import pandas as pd
 import streamlit as st
-from my_model.tabs.run_inference import run_inference
+from my_model.tabs.run_inference import InferenceRunner
 
 
 class UIManager:
@@ -46,7 +46,8 @@ class UIManager:
         st.write("This is a Place Holder until the contents are uploaded.")
 
     def display_run_inference(self):
-        run_inference()
+        inference_runner = InferenceRunner()
+        inference_runner.run_inference()
 
     def display_dissertation_report(self):
         st.title("Dissertation Report")