# -*- coding: utf-8 -*-
"""planets-predictions.ipynb

Automatically generated by Colaboratory.

Original file is located at
    https://colab.research.google.com/drive/1B7jPgW69uJ1nLSAZ2eKfhGx1TSEiYOZh
"""

from fastai.vision.all import *
import gradio as gr

learn = load_learner("planets-v1.pkl")

labels = learn.dls.vocab


def predict(img):
    img = PILImage.create(img)
    pred, pred_idx, probs = learn.predict(img)
    return dict(zip(labels, probs.tolist()))


image_input = gr.Image(shape=(224, 224))
label_output = gr.Label()
examples = [
    "Mercury.jpg",
    "Venus.png",
    "Earth.jpg",
    "Mars.jpg",
    "Jupiter.jpg",
    "Saturn.jpg",
    "Uranus.jpg",
    "Neptune.jpg",
]

iface = gr.Interface(
    fn=predict, inputs=image_input, outputs=label_output, examples=examples
)

iface.launch(inline=False)