vaishnavi kamath commited on
Commit
05e47d5
1 Parent(s): e01ee92

fixed sequence and added gitignore

Browse files
Files changed (2) hide show
  1. .gitignore +2 -0
  2. app.py +1 -1
.gitignore ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ /Model_Data/*
2
+ fastai*.ipynb
app.py CHANGED
@@ -4,7 +4,7 @@ from fastai.vision.all import *
4
 
5
  learn_inf = load_learner('model.pkl')
6
 
7
- categories=('Realism','Impressionism','Cubism', 'Baroque', 'Abstract Expressionism')
8
 
9
  def classify_image(img):
10
  pred,idx,probs=learn_inf.predict(img)
 
4
 
5
  learn_inf = load_learner('model.pkl')
6
 
7
+ categories=('Abstract Expressionism', 'Baroque', 'Cubism', 'Impressionism', 'Realism')
8
 
9
  def classify_image(img):
10
  pred,idx,probs=learn_inf.predict(img)