added stuff

gr.py

@@ -0,0 +1,26 @@
+# %%
+import gradio as gr
+import tensorflow as tf
+import numpy as np
+import cv2
+# %%
+model_folder = 'plants_disease'
+with open(f'{model_folder}/labels.txt', 'r') as f:
+  labels = f.read().split('\n')
+
+# model = tf.saved_model.load(f'{model_folder}/last_layer.hdf5')
+model = tf.keras.models.load_model(f'{model_folder}/last_layer.hdf5')
+# %%
+def classify_image(model, inp):
+  inp = cv2.resize(inp, (224,224,))
+  inp = inp.reshape((-1, 224, 224, 3))
+  inp = tf.keras.applications.mobilenet_v2.preprocess_input(inp)
+  prediction = model.predict(inp).flatten()
+  print(prediction)
+  confidences = {labels[i]: float(prediction[i]) for i in range(1000)}
+  return confidences
+
+gr.Interface(fn=classify_image, 
+             inputs=gr.Image(shape=(224, 224)),
+             outputs=gr.Label(num_top_classes=3),
+             examples=["TomatoHealthy2.jpg", "TomatoYellowCurlVirus3.jpg"]).launch()

requirements.txt

@@ -0,0 +1,99 @@
+absl-py==1.4.0
+aiofiles==23.1.0
+aiohttp==3.8.4
+aiosignal==1.3.1
+altair==5.0.0
+anyio==3.6.2
+astunparse==1.6.3
+async-timeout==4.0.2
+attrs==23.1.0
+cachetools==5.3.0
+certifi==2023.5.7
+charset-normalizer==3.1.0
+click==8.1.3
+contourpy==1.0.7
+cycler==0.11.0
+fastapi==0.95.2
+ffmpy==0.3.0
+filelock==3.12.0
+flatbuffers==23.5.9
+fonttools==4.39.4
+frozenlist==1.3.3
+fsspec==2023.5.0
+gast==0.4.0
+google-auth==2.18.1
+google-auth-oauthlib==1.0.0
+google-pasta==0.2.0
+gradio==3.31.0
+gradio_client==0.2.5
+grpcio==1.54.2
+h11==0.14.0
+h5py==3.8.0
+httpcore==0.17.1
+httpx==0.24.1
+huggingface-hub==0.14.1
+idna==3.4
+importlib-metadata==6.6.0
+importlib-resources==5.12.0
+jax==0.4.10
+Jinja2==3.1.2
+jsonschema==4.17.3
+keras==2.12.0
+kiwisolver==1.4.4
+libclang==16.0.0
+linkify-it-py==2.0.2
+Markdown==3.4.3
+markdown-it-py==2.2.0
+MarkupSafe==2.1.2
+matplotlib==3.7.1
+mdit-py-plugins==0.3.3
+mdurl==0.1.2
+ml-dtypes==0.1.0
+multidict==6.0.4
+numpy==1.23.5
+oauthlib==3.2.2
+opencv-python==4.7.0.72
+opt-einsum==3.3.0
+orjson==3.8.12
+packaging==23.1
+pandas==2.0.1
+Pillow==9.5.0
+pkgutil_resolve_name==1.3.10
+protobuf==4.23.1
+pyasn1==0.5.0
+pyasn1-modules==0.3.0
+pydantic==1.10.7
+pydub==0.25.1
+Pygments==2.15.1
+pyparsing==3.0.9
+pyrsistent==0.19.3
+python-dateutil==2.8.2
+python-multipart==0.0.6
+pytz==2023.3
+PyYAML==6.0
+requests==2.30.0
+requests-oauthlib==1.3.1
+rsa==4.9
+scipy==1.10.1
+semantic-version==2.10.0
+six==1.16.0
+sniffio==1.3.0
+starlette==0.27.0
+tensorboard==2.12.3
+tensorboard-data-server==0.7.0
+tensorflow==2.12.0
+tensorflow-estimator==2.12.0
+tensorflow-io-gcs-filesystem==0.32.0
+termcolor==2.3.0
+toolz==0.12.0
+tqdm==4.65.0
+typing_extensions==4.5.0
+tzdata==2023.3
+uc-micro-py==1.0.2
+urllib3==1.26.15
+uvicorn==0.22.0
+websockets==11.0.3
+Werkzeug==2.3.4
+wrapt==1.14.1
+yarl==1.9.2
+zipp==3.15.0