Skip to content

This is the code for 'Build an AI Startup with Pytorch" by Siraj Raval

License

Notifications You must be signed in to change notification settings

llSourcell/Build-an-AI-Startup-with-PyTorch

Repository files navigation

AI Startup with PyTorch

Overview

This is the code for this video on Youtube by Siraj Raval on building an AI Startup with PyTorch. This is the original Dialogflow + Android integration project. This is the colab that I demonstrated in the video where I train the PyTorch model on movie reviews, then convert it to ONNX, then to Tensorflow's expected protobuf format.

#AISTARTUPCHALLENGE

alt text

Due date is May 15 2019 Submit your entry here.

Dependencies

  • Firebase
  • Paypal
  • Tensorflow Lite
  • DialogFlow
  • PyTorch

Instructions

Download Android Studio then import this project. Since I've placed the dependencies in the build.gradle file, it should auto-download them for you, which is awesome. It needs more features, and better integrated dataflow. I could've put more time into it, but I'm on to the next one. The model was too big to upload to GitHub. Place any .pb model into the app/src/main/assets folder and Tensorflow lite will detect it so it can be used for inference.

Credits

So many people. I stand on the shoulders of giants. Respect to all developers.

About

This is the code for 'Build an AI Startup with Pytorch" by Siraj Raval

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published