WIP

app.py

@@ -1,7 +1,23 @@
 import gradio as gr
+import pickle
 
-def greet(name):
-    return "Hello " + name + "!!"
+categories = ['Telecaster', 'Stratocaster', 'Jazzmaster']
+examples = ['stratocaster.jpt', 'telecaster.jpg', 'jazzmaster.jpg']
+image = gr.inputs.Image(shape=(192, 192))
+label = gr.outputs.Label()
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+# Load the trained model from the model.pkl file
+with open("model.pkl", "rb") as f:
+    model = pickle.load(f)
+
+
+def predict(image):
+    # image = cv2.resize(image, (224, 224))
+    # image = np.expand_dims(image, axis=0)
+    prediction, idx, probabilities = model.predict(image)
+    return dict(zip(categories, map(float, probabilities)))
+
+
+iface = gr.Interface(fn=predict, inputs=image, outputs=label,
+                     examples=examples, capture_session=True)
+iface.launch()

model.pkl

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

requirements.txt

@@ -0,0 +1,62 @@
+aiohttp==3.8.3
+aiosignal==1.3.1
+altair==4.2.0
+anyio==3.6.2
+async-timeout==4.0.2
+attrs==22.2.0
+certifi==2022.12.7
+charset-normalizer==2.1.1
+click==8.1.3
+contourpy==1.0.6
+cycler==0.11.0
+distlib==0.3.6
+entrypoints==0.4
+fastapi==0.88.0
+ffmpy==0.3.0
+filelock==3.8.2
+fonttools==4.38.0
+frozenlist==1.3.3
+fsspec==2022.11.0
+gradio==3.15.0
+h11==0.14.0
+httpcore==0.16.3
+httpx==0.23.1
+idna==3.4
+Jinja2==3.1.2
+jsonschema==4.17.3
+kiwisolver==1.4.4
+linkify-it-py==1.0.3
+markdown-it-py==2.1.0
+MarkupSafe==2.1.1
+matplotlib==3.6.2
+mdit-py-plugins==0.3.3
+mdurl==0.1.2
+multidict==6.0.3
+numpy==1.24.0
+orjson==3.8.3
+packaging==22.0
+pandas==1.5.2
+Pillow==9.3.0
+platformdirs==2.6.0
+pycryptodome==3.16.0
+pydantic==1.10.2
+pydub==0.25.1
+pyparsing==3.0.9
+pyrsistent==0.19.2
+python-dateutil==2.8.2
+python-multipart==0.0.5
+pytz==2022.7
+PyYAML==6.0
+requests==2.28.1
+rfc3986==1.5.0
+six==1.16.0
+sniffio==1.3.0
+starlette==0.22.0
+toolz==0.12.0
+typing_extensions==4.4.0
+uc-micro-py==1.0.1
+urllib3==1.26.13
+uvicorn==0.20.0
+virtualenv==20.17.1
+websockets==10.4
+yarl==1.8.2

serve.py

@@ -0,0 +1,24 @@
+import time
+import subprocess
+import sys
+
+from watchdog.observers import Observer
+from watchdog.events import FileSystemEventHandler
+
+
+class ScriptReloader(FileSystemEventHandler):
+    def on_modified(self, event):
+        subprocess.run([sys.executable, "app.py"])
+
+
+observer = Observer()
+event_handler = ScriptReloader()
+observer.schedule(event_handler, "app.py", recursive=True)
+observer.start()
+
+try:
+    while True:
+        time.sleep(1)
+except KeyboardInterrupt:
+    observer.stop()
+observer.join()