Change to off/on side shot detector

.ipynb_checkpoints/Untitled-checkpoint.ipynb

@@ -0,0 +1,6 @@
+{
+ "cells": [],
+ "metadata": {},
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

app.py

@@ -3,7 +3,7 @@ from fastai.vision.all import *
 import gradio as gr
 learn = load_learner('model.pkl')
 #|export
-categories = ('person bench pressing','person overhead pressing', 'person deadlifting', 'person lateral raising', 'person bent over row', 'person barbell squatting')
+categories = ('on side', 'off side')
 
 def classify_image(img):
     pred,idx,probs = learn.predict(img)
@@ -11,7 +11,7 @@ def classify_image(img):
 
 image = gr.inputs.Image(shape=(192,192))
 label = gr.outputs.Label()
-examples = ['bench_press.jpg', 'deadlift.jpg', 'lateral_raise.jpg']
+examples = ['off side1.jpg', 'on side_59.jpg', 'off side_2.jpg']
 
 intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
 intf.launch(inline=False)

model.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f45457e0878695141d515640aa37907bab7ee037c545e38969c9bbaa7821e3dd
-size 46986017
+oid sha256:b53ea523fa1d4f0aac386f3c8b37354f3a066f48cb3d885da8b0e1d83f15e675
+size 102885589