Diego Carpintero commited on
Commit
956d1f2
1 Parent(s): 42e90e2

update description

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -2,12 +2,17 @@ import gradio as gr
2
  from fastai.vision.all import *
3
 
4
  title = "Interstellar Classifier"
5
- description = "Built for fast.ai 'Practical Deep Learning'"
 
 
 
 
6
  inputs = gr.components.Image()
7
  outputs = gr.components.Label()
8
  examples = "examples"
9
 
10
- model = load_learner("model/model.pkl")
 
11
  labels = model.dls.vocab
12
 
13
 
 
2
  from fastai.vision.all import *
3
 
4
  title = "Interstellar Classifier"
5
+ description = (
6
+ "A naive Galaxy Classifier built for the fast.ai 'Deep Learning' "
7
+ "course by fine tuning ResNet50 (3 epochs) with a custom dataset "
8
+ "of images (150 per label)."
9
+ )
10
  inputs = gr.components.Image()
11
  outputs = gr.components.Label()
12
  examples = "examples"
13
 
14
+ # model = load_learner("model/model.pkl")
15
+ model = load_learner("export.pkl")
16
  labels = model.dls.vocab
17
 
18