Laks Srini commited on
Commit
b787d29
1 Parent(s): 26ee6ef

Update to be classifier

Browse files
Files changed (3) hide show
  1. app.ipynb +0 -0
  2. app.py +1 -1
  3. model.pkl +2 -2
app.ipynb CHANGED
The diff for this file is too large to render. See raw diff
 
app.py CHANGED
@@ -13,7 +13,7 @@ import gradio as gr
13
  learn = load_learner("model.pkl")
14
 
15
  # %% app.ipynb 5
16
- categories = ("kitchen", "living room")
17
 
18
  def classify_img(img):
19
  pred,idx,probs = learn.predict(img)
 
13
  learn = load_learner("model.pkl")
14
 
15
  # %% app.ipynb 5
16
+ categories = ['Bathroom', 'Bedroom', 'Floor plan', 'Front', 'Kitchen', 'Living room', 'Parking', 'Porch', 'Swimming pool', 'Views', 'Yard']
17
 
18
  def classify_img(img):
19
  pred,idx,probs = learn.predict(img)
model.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a28784a7e62201d0fcb4160306adbb0d95c8d41fcb29d58a0690fd9730e20bc1
3
- size 46962565
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b3fdff1c13aeac79a260ff457afa208699fcb823ca30aa039b5d03f513ba03cb
3
+ size 791545705