Model in private repo

README 2.md

@@ -0,0 +1,12 @@
+---
+title: Ofwat Material Classification Demo
+emoji: 🔥
+colorFrom: green
+colorTo: yellow
+sdk: gradio
+sdk_version: 4.7.1
+app_file: app.py
+pinned: false
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

README 3.md

@@ -0,0 +1,12 @@
+---
+title: Ofwat Defects Classification Demo
+emoji: 😻
+colorFrom: gray
+colorTo: red
+sdk: gradio
+sdk_version: 4.7.1
+app_file: app.py
+pinned: false
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,35 @@
+from turtle import title
+import gradio as gr
+from transformers import pipeline
+from PIL import Image
+
+# Initialize the pipeline with your model
+pipe = pipeline("image-classification", model="SubterraAI/ofwat_cleaner_classification")
+
+def classify_image(image):
+    # Convert the input image to PIL format
+    PIL_image = Image.fromarray(image).convert('RGB')
+    
+    # Classify the image using the pipeline
+    res = pipe(PIL_image)
+    
+    # Extract labels and scores
+    return {dic["label"]: dic["score"] for dic in res}
+
+# Create the Gradio interface
+iface = gr.Interface(
+    classify_image, 
+    "image", 
+    "label", 
+    examples=[
+        ["examples/CS.jpg"], 
+        ["examples/GI.jpg"],
+        ["examples/PP.jpg"],
+        ["examples/RC.jpg"]
+    ],
+    description="Upload an image to classify its material.",
+    title="Cleaner Classification with AI by Subterra"
+)
+
+# Launch the interface
+iface.launch()

flagged/log.csv

@@ -0,0 +1,7 @@
+image,output 0,output 1,flag,username,timestamp
+"{""path"":""flagged/image/d97b9ae054eab1bd10dc/PP.jpg"",""url"":""http://127.0.0.1:7860/file=/private/var/folders/5q/yl8pmxm116g6r3k8fd9gk74m0000gn/T/gradio/12885a897d79141c54ccb542582449159242eaa0/PP.jpg"",""size"":null,""orig_name"":""PP.jpg"",""mime_type"":null}","{""label"":""VC"",""confidences"":[{""label"":""VC"",""confidence"":0.7706952691078186},{""label"":""PVC"",""confidence"":0.20648105442523956},{""label"":""PE"",""confidence"":0.005161861423403025},{""label"":""PF"",""confidence"":0.004196972120553255},{""label"":""CO"",""confidence"":0.0034184197429567575}]}","| Prefix | Full Name |
+| ------ | --------- |
+| CS | Carbon Steel |
+| GI | Galvanized Iron |
+| PP | Polypropylene |
+| RC | Reinforced Concrete |",,,2023-11-27 15:26:38.548865

requirements.txt

@@ -0,0 +1,96 @@
+accelerate==0.24.1
+aiofiles==23.2.1
+aiohttp==3.9.0
+aiosignal==1.3.1
+altair==5.1.2
+annotated-types==0.6.0
+anyio==3.7.1
+async-timeout==4.0.3
+attrs==23.1.0
+certifi==2023.11.17
+charset-normalizer==3.3.2
+click==8.1.7
+colorama==0.4.6
+contourpy==1.2.0
+cycler==0.12.1
+datasets==2.4.0
+dill==0.3.5.1
+evaluate==0.4.1
+exceptiongroup==1.2.0
+ExifRead==3.0.0
+fastapi==0.104.1
+ffmpy==0.3.1
+filelock==3.13.1
+fonttools==4.45.1
+frozenlist==1.4.0
+fsspec==2023.10.0
+gradio==4.7.1
+gradio_client==0.7.0
+h11==0.14.0
+httpcore==1.0.2
+httpx==0.25.2
+huggingface-hub==0.19.4
+idna==3.4
+importlib-resources==6.1.1
+Jinja2==3.1.2
+joblib==1.3.2
+jsonschema==4.20.0
+jsonschema-specifications==2023.11.1
+kiwisolver==1.4.5
+markdown-it-py==3.0.0
+MarkupSafe==2.1.3
+matplotlib==3.8.2
+mdurl==0.1.2
+mpmath==1.3.0
+multidict==6.0.4
+multiprocess==0.70.13
+networkx==3.2.1
+numpy==1.26.2
+orjson==3.9.10
+packaging==23.2
+pandas==2.1.3
+Pillow==10.1.0
+psutil==5.9.6
+pyarrow==14.0.1
+pyarrow-hotfix==0.6
+pydantic==2.5.2
+pydantic_core==2.14.5
+pydub==0.25.1
+Pygments==2.17.2
+pyparsing==3.1.1
+python-dateutil==2.8.2
+python-multipart==0.0.6
+pytz==2023.3.post1
+PyYAML==6.0.1
+referencing==0.31.0
+regex==2023.10.3
+requests==2.31.0
+responses==0.18.0
+rich==13.7.0
+rpds-py==0.13.1
+safetensors==0.4.0
+scikit-learn==1.3.2
+scipy==1.11.4
+semantic-version==2.10.0
+shellingham==1.5.4
+six==1.16.0
+sniffio==1.3.0
+starlette==0.27.0
+sympy==1.12
+threadpoolctl==3.2.0
+tokenizers==0.15.0
+tomlkit==0.12.0
+toolz==0.12.0
+torch==2.1.1
+torchvision==0.16.1
+tqdm==4.66.1
+transformers @ git+https://github.com/huggingface/transformers@c5be38cd27bee92be73c73ba09aec8bedf841423
+typer==0.9.0
+typing_extensions==4.8.0
+tzdata==2023.3
+urllib3==2.1.0
+uvicorn==0.24.0.post1
+websockets==11.0.3
+xxhash==3.4.1
+yarl==1.9.3
+zipp==3.17.0