testing speed without input image output

app.py

@@ -150,7 +150,7 @@ def predict_and_analyze(model_name, num_channels, dim, image):
     
     print("Sending to Hugging Face")
     
-    return output, input_pil_image, fig
+    return output, fig
 
 
 if __name__ == "__main__":
@@ -171,7 +171,7 @@ if __name__ == "__main__":
                             show_label=True), 
                 gr.File(label="Input Data", show_label=True)],
         outputs=[gr.Textbox(lines=1, label="Prediction", show_label=True), 
-                gr.Image(label="Input Image", show_label=True), 
+                # gr.Image(label="Input Image", show_label=True), 
                 #  gr.Image(label="Activation 1", show_label=True), 
                 #  gr.Image(label="Actication 2", show_label=True)],
                 gr.Plot(label="Activations", show_label=True)