edits app + req

app.py

@@ -2,11 +2,11 @@
 import gradio as gr
 import torch
 
-###############
+#############
 
 def yolov7_inference(
-    image: gr.inputs.Image = None,
-    conf_threshold: gr.inputs.Slider = 0.50,
+    image: gr.Image = None,
+    conf_threshold: gr.Slider = 0.20,
 ):
 
     device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
@@ -16,14 +16,13 @@ def yolov7_inference(
     results = model([image], size=640)
     return results.render()[0]
         
-
 inputs = [
-    gr.inputs.Image(shape=(640, 480), type="pil", label="Input Image"),
-    gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.50, step=0.05, label="Confidence Threshold"),
+    gr.Image(type="filepath", label="Input"),
+    gr.Slider(minimum=0.0, maximum=1.0, value=0.2, step=0.05, label="Confidence Threshold", interactive=True),
 ]
 
 outputs = [
-    gr.outputs.Image(type="filepath").style(height=400, width=400),
+    gr.Image(type="filepath"),
     
 ]
 
@@ -44,9 +43,13 @@ Email: rrighart@googlemail.com
 Web: <a href="https://www.rrighart.com">www.rrighart.com</a> 
 """,
     article = "<p style='text-align: center'><a href='https://www.rrighart.com' target='_blank'>Webpage</a></p> <p style='text-align: center'><a href='https://www.kaggle.com/code/rrighart/detection-of-product-defects-using-yolov7' target='_blank'>Kaggle</a></p>",
-    examples=[['example1.JPG', 0.50], ['example2.JPG', 0.50], ['example3.JPG', 0.50]],
+    examples = [['example1.JPG'], ['example2.JPG'], ['example3.JPG']],
+    #examples = [['example1.JPG', 0.50], ['example2.JPG', 0.50], ['example3.JPG', 0.50]],
     css=css,
-    cache_examples=True,
+    cache_examples=False,
 )
-demo_app.launch(debug=False, enable_queue=True)
+
+demo_app.queue()
+
+demo_app.launch(debug=False)
 

requirements.txt

@@ -1,151 +1,121 @@
-absl-py==1.2.0
-aiofiles==22.1.0
-aiohttp==3.8.1
-aiosignal==1.2.0
-altair==4.2.0
-analytics-python==1.4.0
-anyio==3.6.1
-asttokens==2.2.1
-astunparse==1.6.3
-async-timeout==4.0.2
-attrs==22.1.0
+absl-py==2.0.0
+aiofiles==23.2.1
+altair==5.2.0
+annotated-types==0.6.0
+anyio==3.7.1
+asttokens==2.4.1
+attrs==23.1.0
 backcall==0.2.0
-backoff==1.10.0
-backports.zoneinfo==0.2.1
-bcrypt==4.0.0
-blinker==1.5
-cachetools==5.2.0
-certifi==2022.6.15
-cffi==1.15.1
-charset-normalizer==2.1.0
-click==8.1.3
-commonmark==0.9.1
-config==0.5.1
-cryptography==37.0.4
-cv==1.0.0
-cycler==0.11.0
+cachetools==5.3.2
+certifi==2023.11.17
+charset-normalizer==3.3.2
+click==8.1.7
+colorama==0.4.6
+contourpy==1.1.1
+cycler==0.12.1
 decorator==5.1.1
-entrypoints==0.4
-executing==1.2.0
-fastapi==0.79.0
-ffmpy==0.3.0
-flatbuffers==1.12
-fonttools==4.37.1
-frozenlist==1.3.1
-fsspec==2022.8.2
-gast==0.4.0
-gitdb==4.0.9
-GitPython==3.1.27
-google-auth==2.10.0
-google-auth-oauthlib==0.4.6
-google-pasta==0.2.0
-gradio==3.17.0
-grpcio==1.47.0
-h11==0.12.0
-h5py==3.7.0
-httpcore==0.15.0
-httpx==0.23.0
-idna==3.3
-importlib-metadata==4.12.0
-importlib-resources==5.9.0
-inference==0.1
-ipython==8.9.0
-jedi==0.18.2
+exceptiongroup==1.2.0
+executing==2.0.1
+fastapi==0.104.1
+ffmpy==0.3.1
+filelock==3.13.1
+fonttools==4.46.0
+fsspec==2023.12.1
+google-auth==2.25.2
+google-auth-oauthlib==1.0.0
+gradio==4.8.0
+gradio-client==0.7.1
+grpcio==1.60.0
+h11==0.14.0
+httpcore==1.0.2
+httpx==0.25.2
+huggingface-hub==0.19.4
+idna==3.6
+importlib-metadata==7.0.0
+importlib-resources==6.1.1
+ipython==8.12.3
+jedi==0.19.1
 Jinja2==3.1.2
-jsonschema==4.14.0
-keras==2.9.0
-Keras-Preprocessing==1.1.2
-kiwisolver==1.4.4
-libclang==14.0.6
-linkify-it-py==1.0.3
-Markdown==3.4.1
-markdown-it-py==2.1.0
-MarkupSafe==2.1.1
-matplotlib==3.5.3
+jsonschema==4.20.0
+jsonschema-specifications==2023.11.2
+kiwisolver==1.4.5
+Markdown==3.5.1
+markdown-it-py==3.0.0
+MarkupSafe==2.1.3
+matplotlib==3.7.4
 matplotlib-inline==0.1.6
-mdit-py-plugins==0.3.0
 mdurl==0.1.2
-monotonic==1.6
-multidict==6.0.2
-numpy==1.23.1
-nvidia-cublas-cu11==11.10.3.66
-nvidia-cuda-nvrtc-cu11==11.7.99
-nvidia-cuda-runtime-cu11==11.7.99
-nvidia-cudnn-cu11==8.5.0.96
-oauthlib==3.2.0
-opencv-python==4.6.0.66
-opt-einsum==3.3.0
-orjson==3.8.0
-packaging==21.3
-pandas==1.4.3
-paramiko==2.11.0
+mpmath==1.3.0
+networkx==3.1
+numpy==1.23.5
+nvidia-cublas-cu12==12.1.3.1
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cudnn-cu12==8.9.2.26
+nvidia-cufft-cu12==11.0.2.54
+nvidia-curand-cu12==10.3.2.106
+nvidia-cusolver-cu12==11.4.5.107
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-nccl-cu12==2.18.1
+nvidia-nvjitlink-cu12==12.3.101
+nvidia-nvtx-cu12==12.1.105
+oauthlib==3.2.2
+opencv-python==4.8.1.78
+orjson==3.9.10
+packaging==23.2
+pandas==2.0.3
 parso==0.8.3
-pexpect==4.8.0
+pexpect==4.9.0
 pickleshare==0.7.5
-Pillow==9.2.0
+Pillow==10.1.0
 pkgutil-resolve-name==1.3.10
-prompt-toolkit==3.0.36
-protobuf==3.19.4
-psutil==5.9.4
+prompt-toolkit==3.0.41
+protobuf==4.21.2
+psutil==5.9.6
 ptyprocess==0.7.0
 pure-eval==0.2.2
-pyarrow==9.0.0
-pyasn1==0.4.8
-pyasn1-modules==0.2.8
-pycparser==2.21
-pycryptodome==3.15.0
-pydantic==1.9.1
-pydeck==0.8.0b1
+pyasn1==0.5.1
+pyasn1-modules==0.3.0
+pydantic==2.5.2
+pydantic-core==2.14.5
 pydub==0.25.1
-Pygments==2.13.0
-Pympler==1.0.1
-PyNaCl==1.5.0
-pyparsing==3.0.9
-pyrsistent==0.18.1
+pygments==2.17.2
+pyparsing==3.1.1
 python-dateutil==2.8.2
-python-multipart==0.0.5
-pytz==2022.2.1
-pytz-deprecation-shim==0.1.0.post0
-PyYAML==6.0
-requests==2.28.1
+python-multipart==0.0.6
+pytz==2023.3.post1
+PyYAML==6.0.1
+referencing==0.32.0
+requests==2.31.0
 requests-oauthlib==1.3.1
-rfc3986==1.5.0
-rich==12.5.1
+rich==13.7.0
+rpds-py==0.13.2
 rsa==4.9
-scipy==1.10.0
-seaborn==0.12.2
-semver==2.13.0
+scipy==1.10.1
+seaborn==0.13.0
+semantic-version==2.10.0
+shellingham==1.5.4
 six==1.16.0
-smmap==5.0.0
-sniffio==1.2.0
-stack-data==0.6.2
-starlette==0.19.1
-streamlit==1.12.2
-tensorboard==2.9.1
-tensorboard-data-server==0.6.1
-tensorboard-plugin-wit==1.8.1
-tensorflow==2.9.1
-tensorflow-estimator==2.9.0
-tensorflow-io-gcs-filesystem==0.26.0
-termcolor==1.1.0
-toml==0.10.2
+sniffio==1.3.0
+stack-data==0.6.3
+starlette==0.27.0
+sympy==1.12
+tensorboard==2.14.0
+tensorboard-data-server==0.7.2
+tomlkit==0.12.0
 toolz==0.12.0
-torch==1.13.1
-torchvision==0.14.1
-tornado==6.2
-tqdm==4.64.1
-traitlets==5.9.0
-typing-extensions==4.3.0
-tzdata==2022.2
-tzlocal==4.2
-uc-micro-py==1.0.1
-urllib3==1.26.11
-uvicorn==0.18.2
-validators==0.20.0
-watchdog==2.1.9
-wcwidth==0.2.6
-websockets==10.3
-Werkzeug==2.2.2
-wrapt==1.14.1
-yarl==1.8.1
-zipp==3.8.1
+torch==2.1.1
+torchvision==0.16.1
+tqdm==4.66.1
+traitlets==5.14.0
+triton==2.1.0
+typer==0.9.0
+typing-extensions==4.8.0
+tzdata==2023.3
+urllib3==2.1.0
+uvicorn==0.24.0.post1
+wcwidth==0.2.12
+websockets==11.0.3
+werkzeug==3.0.1
+zipp==3.17.0