AItool commited on
Commit
cc2bd42
·
verified ·
1 Parent(s): ecd3502

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -6,7 +6,6 @@ __all__ = ['learn', 'classify_image', 'categories', 'classifier', 'dog','image',
6
  from fastai.vision.all import *
7
  import gradio as gr
8
  import timm
9
- import streamlit as st
10
 
11
  # Cell
12
  learn = load_learner('model.pkl')
@@ -19,7 +18,7 @@ def classify_image(img):
19
  return dict(zip(categories, map(float,probs)))
20
 
21
  # Cell
22
- st.title("Chihuahua,Winner or Basset Hound")
23
  image = gr.inputs.Image(shape=(192, 192))
24
  label = gr.outputs.Label()
25
  examples = ['dog0.jpg','dog1.jpg','dog2.jpg','dog3.jpg','dog4.jpg','dog5.jpg','dog6.jpg']
 
6
  from fastai.vision.all import *
7
  import gradio as gr
8
  import timm
 
9
 
10
  # Cell
11
  learn = load_learner('model.pkl')
 
18
  return dict(zip(categories, map(float,probs)))
19
 
20
  # Cell
21
+ #st.title("Chihuahua,Winner or Basset Hound")
22
  image = gr.inputs.Image(shape=(192, 192))
23
  label = gr.outputs.Label()
24
  examples = ['dog0.jpg','dog1.jpg','dog2.jpg','dog3.jpg','dog4.jpg','dog5.jpg','dog6.jpg']