classifier app

.ipynb_checkpoints/classifier-checkpoint.ipynb

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

app.py

@@ -1,7 +1,31 @@
 import gradio as gr
+import numpy as np
 
-def greet(name):
-    return "Hello " + name + "!!"
+# Load a pre-trained image classification model
+model = tf.keras.applications.MobileNetV2(weights='imagenet')
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+# Function to make predictions from an image
+def classify_image(image):
+    # Resize the image to the required input size of the model
+    
+    # Preprocess the image to match the model's requirements
+    
+    # Make a prediction
+    
+    # Decode the prediction and get the class name
+    class_name = np.random.randint(0,3)
+    classes = ["cat", "dog", "flower"]
+    return classes[class_name]
+
+# Sample images for user to choose from
+sample_images = ["cat.jpg", "dog.jpg", "flower.jpg"]
+
+iface = gr.Interface(
+    fn=classify_image,
+    inputs=gr.inputs.Image(label="Select an image", type="file"),
+    outputs="text",
+    live=True,
+    examples=sample_images
+)
+
+iface.launch()

classifier.ipynb

@@ -0,0 +1,54 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "id": "6afcec5d",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import fastbook\n",
+    "fastbook.setup_book()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "id": "4e4f004b",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from fastai.vision.all import *"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "a57b7136",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python [conda env:hf]",
+   "language": "python",
+   "name": "conda-env-hf-py"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.11.6"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

data.zip

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