Spaces:

louiecerv
/

html_javascript_python_CNN_webapp

Running

App Files Files Community

louiecerv commited on Feb 7, 2025

Commit

59ff5b7

verified ·

1 Parent(s): 66e1fbb

Update index.html

Browse files

Files changed (1) hide show

index.html +57 -22

index.html CHANGED Viewed

@@ -1,29 +1,64 @@
 <!DOCTYPE html>
 <html lang="en">
 <head>
-    <meta charset="UTF-8" />
-    <link rel="stylesheet" href="style.css" />
-    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
-    <title>Transformers.js - Object Detection</title>
 </head>
 <body>
-    <h1>Object Detection w/ 🤗 Transformers.js</h1>
-    <label id="container" for="upload">
-        <svg width="25" height="25" viewBox="0 0 25 25" fill="none" xmlns="http://www.w3.org/2000/svg">
-            <path fill="#000"
-                d="M3.5 24.3a3 3 0 0 1-1.9-.8c-.5-.5-.8-1.2-.8-1.9V2.9c0-.7.3-1.3.8-1.9.6-.5 1.2-.7 2-.7h18.6c.7 0 1.3.2 1.9.7.5.6.7 1.2.7 2v18.6c0 .7-.2 1.4-.7 1.9a3 3 0 0 1-2 .8H3.6Zm0-2.7h18.7V2.9H3.5v18.7Zm2.7-2.7h13.3c.3 0 .5 0 .6-.3v-.7l-3.7-5a.6.6 0 0 0-.6-.2c-.2 0-.4 0-.5.3l-3.5 4.6-2.4-3.3a.6.6 0 0 0-.6-.3c-.2 0-.4.1-.5.3l-2.7 3.6c-.1.2-.2.4 0 .7.1.2.3.3.6.3Z">
-            </path>
-        </svg>
-        Click to upload image
-        <label id="example">(or try example)</label>
-    </label>
-    <label id="status">Loading model...</label>
-    <input id="upload" type="file" accept="image/*" />
-    <script src="index.js" type="module"></script>
-</body>
-</html>

 <!DOCTYPE html>
 <html lang="en">
 <head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Cat vs. Dog Image Classifier</title>
+    <script src="https://cdn.jsdelivr.net/npm/@xenova/transformers"></script>
+    <style>
+        body {
+            font-family: Arial, sans-serif;
+            text-align: center;
+            margin: 50px;
+        }
+        img {
+            max-width: 300px;
+            margin: 20px;
+        }
+    </style>
 </head>
 <body>
+    <h1>Cat vs. Dog Image Classifier</h1>
+    <input type="file" id="imageUploader" accept="image/*">
+    <br>
+    <img id="uploadedImage" style="display: none;" />
+    <br>
+    <button onclick="predictImage()">Predict</button>
+    <h2 id="result"></h2>
+    <script>
+        let model;
+        async function loadModel() {
+            model = await transformers.pipeline('image-classification', 'louiecerv/cats_dogs_recognition_tf_cnn');
+            console.log("Model loaded successfully");
+        }
+        loadModel();
+        document.getElementById('imageUploader').addEventListener('change', function(event) {
+            const file = event.target.files[0];
+            if (file) {
+                const reader = new FileReader();
+                reader.onload = function(e) {
+                    const imgElement = document.getElementById('uploadedImage');
+                    imgElement.src = e.target.result;
+                    imgElement.style.display = 'block';
+                };
+                reader.readAsDataURL(file);
+            }
+        });
+        async function predictImage() {
+            const imgElement = document.getElementById('uploadedImage');
+            if (!imgElement.src) {
+                alert("Please upload an image first.");
+                return;
+            }
+            const predictions = await model(imgElement);
+            const topPrediction = predictions[0];
+            document.getElementById('result').innerText = `Prediction: ${topPrediction.label} (Confidence: ${(topPrediction.score * 100).toFixed(2)}%)`;
+        }
+    </script>
+</body>
+</html>