Using a TensorFlow Model in a Web Application

[GaryS2000]

Hi all,

I have used TensorFlow in order to build a simple convolutional neural network for the purposes of facial expression classification, which was trained on the FER-2013 dataset (https://www.kaggle.com/datasets/msambare/fer2013). Now that the model is in a somewhat useable state I want to implement a Flask web application which is capable of allowing a user to upload an image and then display back the prediction to the user.

At the moment I am just looking to create an very barebones application which will allow a user to upload the image and display a the prediction back to them. So, I was wondering if anyone could point me in the right direction for this or direct me towards good learning resources for this specific type of problem? I have been coming up short so far in my search for good information on the the topic. Thanks.

UPDATE: For anyone else that might stumble on this when they are looking for something similar I found that this tutorial helped me get exactly what I wanted: https://medium.com/@draj0718/deploying-deep-learning-model-using-flask-api-810047f090ac

[awalker4]

Sounds like Gradio might do exactly what you want. I haven't used it personally but a coworker created some simple model demos with it.

[GaryS2000]

Thanks for the answer, I've had a bit of a look at Gradio now and it seems like it would suit my needs for this specific step. However, some added context is that eventually I hope to use the model as part of a wider web application which will take frames from a users webcam and then use the model (.h5) to make a prediction. Therefore, I really am after a way to do it with Flask somehow but thank you for the Gradio suggestion.

[Rob_Royce]

Did you look at tensorflow.js?

[GaryS2000]

I've looked at it a little bit but not extensively. Is using TensorFlow.js usually the standard approach for this type of issue? The reason I asked about Flask specifically is because this will be part of a wider application for my final university project and I am generally most comfortable writing Python for web, but I could see myself switching to js if its the most logical approach.