Spaces:
Sleeping
Sleeping
Dan Biagini
commited on
Commit
·
5b18dae
1
Parent(s):
9f50c5e
center description text
Browse files
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')
|