try again: westworld ML files

app.py

@@ -0,0 +1,12 @@
+from fastai.vision.all import *
+import gradio as gr
+title = 'Westworld host detector'
+description = 'Avoid being killed by a host disguised as a human using this detector'
+learn = load_learner('westworld2.pkl')
+labels = learn.dls.vocab
+def predict(img):
+    img = PILImage.create(img)
+    pred,pred_idx,probs = learn.predict(img)
+    return {labels[i]: float(probs[i]) for i in range(len(labels))}
+examples = [['dolores.jpg'], ['hotdog.jpg']]
+gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)), examples=examples, title=title, description=description, outputs=gr.outputs.Label(num_top_classes=3)).launch(share=True)

requirements.txt

@@ -0,0 +1 @@
+fastai

westworld2.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d96564a173571815bc02507cc77d9a653a07ef4bf93058ea9f4e7920343ddc55
+size 46963361