Dan Biagini commited on
Commit
5b18dae
·
1 Parent(s): 9f50c5e

center description text

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -5,11 +5,12 @@ import random
5
 
6
  title = "Who Am I? (Hockey Edition)"
7
 
8
- desc = """This app uses a 'neural network' (a type of machine learning model) to classify
9
- an image as containing a hockey player, a hockey goalie or a hockey referee."""
10
 
11
  art = """I built this model using about 50 hockey related images found on the web and in my own collection. I started with a pretrained `resnet18` model (resnet18 is trained on `imagenet`, a very large dataset with millions of images)
12
  and then performed `fine tuning` of this using python and the `fast.ai` library.
 
13
  The total training time for this was about 5 minutes on a basic GPU. It's impressive how accurate this quick / small model can be!"""
14
 
15
  learn = load_learner('hockey_model.pkl')
 
5
 
6
  title = "Who Am I? (Hockey Edition)"
7
 
8
+ desc = """<center>This app uses a 'neural network' (a type of machine learning model) to classify
9
+ an image as containing a hockey player, a hockey goalie or a hockey referee.</center>"""
10
 
11
  art = """I built this model using about 50 hockey related images found on the web and in my own collection. I started with a pretrained `resnet18` model (resnet18 is trained on `imagenet`, a very large dataset with millions of images)
12
  and then performed `fine tuning` of this using python and the `fast.ai` library.
13
+
14
  The total training time for this was about 5 minutes on a basic GPU. It's impressive how accurate this quick / small model can be!"""
15
 
16
  learn = load_learner('hockey_model.pkl')