Spaces:

ak0601
/

cat_dog_classifier

Runtime error

ak0601 commited on Dec 26, 2023

Commit

92dc739

•

1 Parent(s): 6c8d6ce

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -30,6 +30,6 @@ def predict(image):
     return confidences, infer
 gr.Interface(fn=predict,
-             inputs=gr.Image(shape=(512, 512)),
              outputs=[gr.outputs.Label(num_top_classes=3), gr.outputs.Textbox('infer',label='Inference Time')],
              examples='1.jpg 2.jpg'.split(' ')).launch()

     return confidences, infer
 gr.Interface(fn=predict,
+             inputs=gr.Image(),
              outputs=[gr.outputs.Label(num_top_classes=3), gr.outputs.Textbox('infer',label='Inference Time')],
              examples='1.jpg 2.jpg'.split(' ')).launch()