initial upload

app.py

@@ -0,0 +1,29 @@
+from fastcore.all import *
+from fastai.vision.all import *
+import gradio as gr
+
+learn = load_leaner('export.pkl')
+
+def predict(img):
+  img = PILImage.create(img)
+  ctr = learn.predict(img)[0]
+  
+  img_resized = img.reshape(240, 320)
+  fig = plt.figure()
+  plt.imshow(img_resized)
+  plt.scatter(ctr[0][0], ctr[0][1], color = 'red')
+  plt.axis('off')
+  plt.show()
+  return fig
+
+title = "Face Center"
+description = "This model that is trained to find the center of faces in the uploaded picture."
+
+gr.Interface(
+    fn=predict,
+    inputs='image',
+    outputs='plot',
+    title=title,
+    description=description,
+    # examples = ['scott.png','dwight.jpeg','stanley.jpg']
+).launch(share=False)

export.pkl

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

requirements.txt

@@ -0,0 +1,2 @@
+fastai
+fastcore