input to RGB at end

app.py

@@ -7,9 +7,10 @@ import numpy as np
 # import onnx
 import onnxruntime as ort
 # from onnx import helper
-from optimum.onnxruntime import ORTModel
+# from optimum.onnxruntime import ORTModel
 
-import pandas as pd
+# import pandas as pd
+from PIL import Image
 
 from scipy import special
 
@@ -167,6 +168,9 @@ def predict_and_analyze(model_name, num_channels, dim, image):
     activation_1 = normalize_array(activation_1)
     activation_2 = normalize_array(activation_2)
     
+    # convert input image to RGB
+    input_image = Image.fromarray(np.uint8(cm.magma(input_image)*255))
+    
     print("Plotting")
     
     origin = 'lower'

requirements.txt

@@ -2,8 +2,5 @@ torch
 numpy
 matplotlib
 scipy
-onnx
 onnxruntime
-streamlit
-onnx2pytorch
-optimum
+Pillow