Deepak107 commited on
Commit
3b16089
·
1 Parent(s): 751c052

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ def predict_input_image(img):
9
  return {class_names[i]: float(prediction[i]) for i in range(len(class_names))}
10
 
11
 
12
- image = gr.inputs.Image(shape=(254,254))
13
  label = gr.outputs.Label(num_top_classes=len(class_names))
14
 
15
  gr.Interface(fn=predict_input_image, inputs=image, outputs=label,interpretation='default').launch(debug='True')
 
9
  return {class_names[i]: float(prediction[i]) for i in range(len(class_names))}
10
 
11
 
12
+ image = gr.inputs.Image(shape=(224,224))
13
  label = gr.outputs.Label(num_top_classes=len(class_names))
14
 
15
  gr.Interface(fn=predict_input_image, inputs=image, outputs=label,interpretation='default').launch(debug='True')