Initial app

README.md

@@ -1,12 +1,13 @@
 ---
 title: Traffic
-emoji: 📈
+emoji: 🚦
 colorFrom: red
-colorTo: pink
+colorTo: yellow
 sdk: gradio
 sdk_version: 4.7.1
 app_file: app.py
-pinned: false
+pinned: true
+license: apache-2.0
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,92 @@
+import os
+import gradio as gr
+from huggingface_hub import login
+from huggingface_hub import snapshot_download
+import numpy as np
+import tensorflow as tf
+import cv2
+
+IMG_WIDTH = 32
+IMG_HEIGHT = 32
+
+classes = { 0:'Speed limit (20km/h)',
+            1:'Speed limit (30km/h)', 
+            2:'Speed limit (50km/h)', 
+            3:'Speed limit (60km/h)', 
+            4:'Speed limit (70km/h)', 
+            5:'Speed limit (80km/h)', 
+            6:'End of speed limit (80km/h)', 
+            7:'Speed limit (100km/h)', 
+            8:'Speed limit (120km/h)', 
+            9:'No passing', 
+            10:'No passing veh over 3.5 tons', 
+            11:'Right-of-way at intersection', 
+            12:'Priority road', 
+            13:'Yield', 
+            14:'Stop', 
+            15:'No vehicles', 
+            16:'Veh > 3.5 tons prohibited', 
+            17:'No entry', 
+            18:'General caution', 
+            19:'Dangerous curve left', 
+            20:'Dangerous curve right', 
+            21:'Double curve', 
+            22:'Bumpy road', 
+            23:'Slippery road', 
+            24:'Road narrows on the right', 
+            25:'Road work', 
+            26:'Traffic signals', 
+            27:'Pedestrians', 
+            28:'Children crossing', 
+            29:'Bicycles crossing', 
+            30:'Beware of ice/snow',
+            31:'Wild animals crossing', 
+            32:'End speed + passing limits', 
+            33:'Turn right ahead', 
+            34:'Turn left ahead', 
+            35:'Ahead only', 
+            36:'Go straight or right', 
+            37:'Go straight or left', 
+            38:'Keep right', 
+            39:'Keep left', 
+            40:'Roundabout mandatory', 
+            41:'End of no passing', 
+            42:'End no passing veh > 3.5 tons' }
+
+def image_mod(image):
+    
+    # Resize image to the dimensions used when training 
+    size = IMG_WIDTH, IMG_HEIGHT
+    res = cv2.resize(image, size, interpolation=cv2.INTER_AREA)
+
+    # Convert image from PIL format (RGB) to the cv2 format
+    # (BGR) that was used when training the model
+    res = cv2.cvtColor(res, cv2.COLOR_RGB2BGR)
+
+    # Convert to float and normalize to match the training
+    res = res.astype("float32") / 255.0
+    
+    # Convert single image to a batch for prediction
+    res = np.array([res])  
+    
+    # Carry out prediction
+    result = model.predict(res)
+
+    # Convert to format suitable for the Label Gradio component
+    confidences = result[0]
+    return {F"{index}: {classes[index]}":element for index, element in enumerate(confidences)}
+
+# Download the model from Hugging Face Hub
+login(token=os.environ['TOKEN_TRAFFIC'])
+model_path = snapshot_download(repo_id=os.environ['REPO_TRAFFIC_MODEL'])
+model = tf.keras.models.load_model(model_path)
+
+# Configure Gradio components
+input_image_component = gr.Image(type="numpy")
+output_label_component = gr.Label(num_top_classes=5)
+
+# Configure user interface
+iface = gr.Interface(fn=image_mod, inputs=input_image_component, outputs=output_label_component, live=True, title="German traffic sign recognizer", description="A convolutional neural network to categorize images of German traffic signs.", article="# Reference\nJ. Stallkamp, M. Schlipsing, J. Salmen, and C. Igel. The German Traffic Sign Recognition Benchmark: A multi-class classification competition. In Proceedings of the IEEE International Joint Conference on Neural Networks, pages 1453–1460. 2011.", examples="examples")
+
+# Launch the frontend server
+iface.launch(share=False)

requirements.txt

@@ -0,0 +1,94 @@
+absl-py==2.0.0
+aiofiles==23.2.1
+altair==5.1.2
+annotated-types==0.6.0
+anyio==3.7.1
+astunparse==1.6.3
+attrs==23.1.0
+cachetools==5.3.2
+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
+fastapi==0.104.1
+ffmpy==0.3.1
+filelock==3.13.1
+flatbuffers==23.5.26
+fonttools==4.44.3
+fsspec==2023.10.0
+gast==0.5.4
+google-auth==2.23.4
+google-auth-oauthlib==1.1.0
+google-pasta==0.2.0
+gradio==4.4.1
+gradio_client==0.7.0
+grpcio==1.59.3
+h11==0.14.0
+h5py==3.10.0
+httpcore==1.0.2
+httpx==0.25.1
+huggingface-hub==0.19.4
+idna==3.4
+importlib-resources==6.1.1
+Jinja2==3.1.2
+jsonschema==4.20.0
+jsonschema-specifications==2023.11.1
+keras==2.15.0
+kiwisolver==1.4.5
+libclang==16.0.6
+Markdown==3.5.1
+markdown-it-py==3.0.0
+MarkupSafe==2.1.3
+matplotlib==3.8.2
+mdurl==0.1.2
+ml-dtypes==0.2.0
+numpy==1.26.2
+oauthlib==3.2.2
+opencv-python==4.8.1.78
+opt-einsum==3.3.0
+orjson==3.9.10
+packaging==23.2
+pandas==2.1.3
+Pillow==10.1.0
+protobuf==4.23.4
+pyasn1==0.5.0
+pyasn1-modules==0.3.0
+pydantic==2.5.1
+pydantic_core==2.14.3
+pydub==0.25.1
+Pygments==2.17.1
+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
+requests==2.31.0
+requests-oauthlib==1.3.1
+rich==13.7.0
+rpds-py==0.13.0
+rsa==4.9
+semantic-version==2.10.0
+shellingham==1.5.4
+six==1.16.0
+sniffio==1.3.0
+starlette==0.27.0
+tensorboard==2.15.1
+tensorboard-data-server==0.7.2
+tensorflow==2.15.0
+tensorflow-estimator==2.15.0
+tensorflow-io-gcs-filesystem==0.31.0
+termcolor==2.3.0
+tomlkit==0.12.0
+toolz==0.12.0
+tqdm==4.66.1
+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
+Werkzeug==3.0.1
+wrapt==1.14.1