Duplicate from RaidedCluster/Sniffusion_PomerAInian

Co-authored-by: Johan S Daniel <RaidedCluster@users.noreply.huggingface.co>

.gitattributes

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,14 @@
+---
+title: Sniffusion PomerAInian
+emoji: 🔥
+colorFrom: pink
+colorTo: gray
+sdk: streamlit
+sdk_version: 1.17.0
+app_file: app.py
+pinned: false
+license: other
+duplicated_from: RaidedCluster/Sniffusion_PomerAInian
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

Sniffusion_PomerAInian.h5

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

app.py

@@ -0,0 +1,35 @@
+from tensorflow.keras.models import load_model
+from huggingface_hub import from_pretrained_keras
+import streamlit as st
+import numpy as np
+import cv2
+from PIL import Image
+
+st.markdown('<img src="https://static.wixstatic.com/media/c2240c_1edbbcded56c4dcbb9cc90a7922021b1~mv2.png/v1/fill/w_333,h_370,al_c,lg_1,q_85,enc_auto/pomai.png" alt="Image" style="width: 200px;">', unsafe_allow_html=True)
+
+st.header("Sniffusion PomerAInian")
+st.write("Human/AI Art Classifier.")
+st.write("NOTE: PomerAInian is a small model and was only trained on LEXICA Stable Diffusion images, images generated by other models may not be classified correctly.")
+upload= st.file_uploader('Insert image for detection:', type=['png','jpg'])
+c1, c2= st.columns(2)
+if upload is not None:
+    im= Image.open(upload)
+    img= np.asarray(im)
+    img = cv2.resize(img, (224, 224))
+    img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+    img = img / 255.0
+    img = np.expand_dims(img, axis=0)
+    c1.header('Input Image')
+    c1.image(im)
+    c1.write(img.shape)
+    model = from_pretrained_keras("RaidedCluster/Sniffusion-PomerAInian")
+    prediction = model.predict(img)
+    hf=str(prediction[0][0]*100)+'% Human Factor'
+    c2.header('Output')
+    c2.subheader('Estimation:')
+    if prediction >=0.5:
+        est="Estimated to be Human Art."
+    else:
+        est="Estimated to be AI Art."
+    c2.write(est)
+    c2.write(hf)

requirements.txt

@@ -0,0 +1,5 @@
+streamlit
+numpy
+opencv-python
+tensorflow
+pillow