update to gradio components

app.py

@@ -3,6 +3,8 @@ from fastai.vision.all import *
 
 title = "Interstellar Classifier"
 description = "Built for fast.ai 'Practical Deep Learning'"
+inputs = gr.components.Image()
+outputs = gr.components.Label()
 examples = "examples"
 
 model = load_learner("model/model.pkl")
@@ -10,15 +12,14 @@ labels = model.dls.vocab
 
 
 def predict(img):
-    img = PILImage.create(img)
     pred, pred_idx, probs = model.predict(img)
     return dict(zip(labels, map(float, probs)))
 
 
 demo = gr.Interface(
     fn=predict,
-    inputs=gr.inputs.Image(shape=(512, 512)),
-    outputs="image",
+    inputs=inputs,
+    outputs=outputs,
     examples=examples,
     title=title,
     description=description,