Spidartist commited on
Commit
01b450d
·
1 Parent(s): 00805ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -42,8 +42,8 @@ def predict_input_image(img):
42
  return {class_names[i]: float(prediction[i]) for i in range(len(class_names))}
43
 
44
 
45
- image = gr.inputs.Image(shape=(224, 224))
46
- label = gr.outputs.Label(num_top_classes=len(class_names))
47
 
48
  gr.Interface(fn=predict_input_image, inputs=image, outputs=label, theme="default",
49
  css=".footer{display:none !important}").launch(debug='True')
 
42
  return {class_names[i]: float(prediction[i]) for i in range(len(class_names))}
43
 
44
 
45
+ image = gr.Image(shape=(224, 224))
46
+ label = gr.Label(num_top_classes=len(class_names))
47
 
48
  gr.Interface(fn=predict_input_image, inputs=image, outputs=label, theme="default",
49
  css=".footer{display:none !important}").launch(debug='True')