Bhavya Giri commited on
Commit
2a366ac
·
1 Parent(s): 78890bf

let's deploy to huggingface spaces

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ def classify_garbage(img):
7
  pred,idx,probs = learn.predict(img)
8
  return(dict(zip(pred,map(float,probs))))
9
 
10
- image = gr.inputs.Image(shape = (224,224))
11
  label = gr.outputs.Label()
12
  title = "Garbage Classifier"
13
  description = "A Garbage classifier trained with fastai. Created as a demo for Gradio and HuggingFace Spaces."
 
7
  pred,idx,probs = learn.predict(img)
8
  return(dict(zip(pred,map(float,probs))))
9
 
10
+ image = gr.inputs.Image(shape = (128,128))
11
  label = gr.outputs.Label()
12
  title = "Garbage Classifier"
13
  description = "A Garbage classifier trained with fastai. Created as a demo for Gradio and HuggingFace Spaces."