add apex to requirements instead of building wheel in project

app.py

@@ -1,10 +1,10 @@
-import pip
+import apex
 import streamlit as st
 from itertools import tee
 from model import InferenceBuilder
 
-pip.main(['install', './apex-0.1-cp311-cp311-linux_x86_64.whl']) # install the apex package from wheel since building takes forever
-import apex
+# pip.main(['install', './apex-0.1-cp311-cp311-linux_x86_64.whl']) # install the apex package from wheel since building takes forever
+# import apex
 
 
 MODEL_AVATAR_URL= "./iphone_robot.png"

model.py

@@ -1,3 +1,4 @@
+import apex
 import mlflow
 import torch
 import streamlit as st

requirements.txt

@@ -1,3 +1,4 @@
+apex
 torch
 mlflow
 transformers