initial commit

README.md

@@ -1,12 +1,12 @@
 ---
 title: MemeDetect
 emoji: 👀
-colorFrom: blue
+colorFrom: red
 colorTo: blue
 sdk: gradio
 sdk_version: 4.16.0
 app_file: app.py
-pinned: false
+pinned: true
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,32 @@
+import gradio as gr
+import tensorflow as tf
+from tensorflow import keras
+from tensorflow.keras.preprocessing.image import img_to_array
+from tensorflow.image import resize
+
+model = keras.models.load_model('memeclf.h5')
+
+
+def predict(image):
+  img = resize(image, (256, 256))
+  img = img_to_array(img)
+  img = tf.expand_dims(img, axis=0)
+  img = tf.cast(img/255. ,tf.float32)
+
+  predictions = model.predict(img)
+  if predictions[0] < 0.5:
+    return "Image is a meme"
+  else:
+    return "Image is not a meme"
+
+
+interface = gr.Interface(
+    fn=predict, 
+    inputs=gr.Image(), 
+    outputs="label",
+    title="Meme Classifier",
+    description="Upload an image and the model will predict if it's a meme or not.",
+    theme="huggingface"
+)
+interface.launch()
+

memeclf.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:921c57daa510a4d8ce701b5fb0faab72f822ac7a0ca7a699de34813605e820ac
+size 178225088

requirements.txt

@@ -0,0 +1,96 @@
+absl-py==2.1.0
+aiofiles==23.2.1
+altair==5.2.0
+annotated-types==0.6.0
+anyio==4.2.0
+astunparse==1.6.3
+attrs==23.2.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.109.0
+ffmpy==0.3.1
+filelock==3.13.1
+flatbuffers==23.5.26
+fonttools==4.47.2
+fsspec==2023.12.2
+gast==0.5.4
+google-auth==2.27.0
+google-auth-oauthlib==1.2.0
+google-pasta==0.2.0
+gradio==4.16.0
+gradio_client==0.8.1
+grpcio==1.60.0
+h11==0.14.0
+h5py==3.10.0
+httpcore==1.0.2
+httpx==0.26.0
+huggingface-hub==0.20.3
+idna==3.6
+importlib-resources==6.1.1
+Jinja2==3.1.3
+jsonschema==4.21.1
+jsonschema-specifications==2023.12.1
+keras==2.15.0
+kiwisolver==1.4.5
+libclang==16.0.6
+Markdown==3.5.2
+markdown-it-py==3.0.0
+MarkupSafe==2.1.4
+matplotlib==3.8.2
+mdurl==0.1.2
+ml-dtypes==0.2.0
+numpy==1.26.3
+oauthlib==3.2.2
+opt-einsum==3.3.0
+orjson==3.9.12
+packaging==23.2
+pandas==2.2.0
+pickles==0.1.1
+pillow==10.2.0
+protobuf==4.23.4
+pyasn1==0.5.1
+pyasn1-modules==0.3.0
+pydantic==2.6.0
+pydantic_core==2.16.1
+pydub==0.25.1
+Pygments==2.17.2
+pyparsing==3.1.1
+python-dateutil==2.8.2
+python-multipart==0.0.6
+pytz==2023.4
+PyYAML==6.0.1
+referencing==0.33.0
+requests==2.31.0
+requests-oauthlib==1.3.1
+rich==13.7.0
+rpds-py==0.17.1
+rsa==4.9
+ruff==0.1.15
+semantic-version==2.10.0
+shellingham==1.5.4
+six==1.16.0
+sniffio==1.3.0
+starlette==0.35.1
+tensorboard==2.15.1
+tensorboard-data-server==0.7.2
+tensorflow==2.15.0
+tensorflow-estimator==2.15.0
+tensorflow-intel==2.15.0
+tensorflow-io-gcs-filesystem==0.31.0
+termcolor==2.4.0
+tomlkit==0.12.0
+toolz==0.12.1
+tqdm==4.66.1
+typer==0.9.0
+typing_extensions==4.9.0
+tzdata==2023.4
+urllib3==2.2.0
+uvicorn==0.27.0.post1
+websockets==11.0.3
+Werkzeug==3.0.1
+wrapt==1.14.1