Skip to content

Instantly share code, notes, and snippets.

View tori29umai0123's full-sized avatar
๐Ÿ 

tori29umai tori29umai0123

๐Ÿ 
View GitHub Profile
@tori29umai0123
tori29umai0123 / AI-NovelChat.py
Last active August 10, 2024 08:05
AI-NovelChat
import os
import sys
import time
import socket
import gradio as gr
from llama_cpp import Llama
import datetime
from jinja2 import Template
import configparser
import threading
@tori29umai0123
tori29umai0123 / rag_stable_diffusion_prompt.py
Last active July 9, 2024 12:36
rag_stable_diffusion_prompt.py
import csv
from pathlib import Path
import re
from sentence_transformers import SentenceTransformer
import faiss
from llama_cpp import Llama
import numpy as np
@tori29umai0123
tori29umai0123 / danbooru_tags.csv
Created July 9, 2024 10:21
danbooru_tags.csv
We can't make this file beautiful and searchable because it's too large.
1girl,4114588
solo,3426446
highres,3008413
long_hair,2898315
commentary_request,2610959
breasts,2252741
looking_at_viewer,2017032
blush,1987494
smile,1847264
short_hair,1530601
import csv
from pathlib import Path
import re
from langchain_core.prompts import ChatPromptTemplate
from langchain.chains import create_retrieval_chain
from langchain.chains.combine_documents import create_stuff_documents_chain
from langchain_community.vectorstores import FAISS
from langchain_community.llms import LlamaCpp
from langchain.prompts import PromptTemplate
from langchain_core.documents import Document
import os
from concurrent.futures import ThreadPoolExecutor
from PIL import Image, ImageFilter
from PIL import Image, ImageFilter
def apply_blur_to_image(image_path, output_path, blur_radius):
"""ๆŒ‡ๅฎšใ•ใ‚ŒใŸ็”ปๅƒใซใƒ–ใƒฉใƒผๅŠนๆžœใ‚’้ฉ็”จใ—ใ€ๆŒ‡ๅฎšใ•ใ‚ŒใŸใƒ‘ใ‚นใซไฟๅญ˜ใ™ใ‚‹"""
try:
image = Image.open(image_path)
#็ทš็”ปใ‚ใ‚Š็”ปๅƒใ‹ใ‚‰ใ€็ทš็”ปใ‚’ๅ–ใ‚Š้™คใใ‚นใ‚ฏใƒชใƒ—ใƒˆใ€‚็ทš็”ปใ‚’ๅ–ใ‚Š้™คใใŸใ„็”ปๅƒ๏ผˆimage_dir๏ผ‰ใŒใ‚ใ‚Œใฐๅ‹•ใใ€‚็ทš็”ปๅ…ฅๅŠ›ไธ่ฆใ€‚
from PIL import Image, ImageFilter, ImageOps
from collections import defaultdict
from skimage import color as sk_color
from tqdm import tqdm
from skimage.color import deltaE_ciede2000, rgb2lab
import cv2
import numpy as np
import os
@tori29umai0123
tori29umai0123 / ContentSafetyAnalyzer.py
Last active June 14, 2024 02:36
ContentSafetyAnalyzer.py
import csv
import os
from pathlib import Path
import cv2
import numpy as np
from PIL import Image
import onnxruntime as ort
from huggingface_hub import hf_hub_download
@tori29umai0123
tori29umai0123 / nsfw_filter_with_tagger.py
Last active June 5, 2024 01:47
nsfw_filter_with_tagger.py
import csv
import glob
import os
from pathlib import Path
import cv2
import numpy as np
import torch
from PIL import Image
from tqdm import tqdm
@tori29umai0123
tori29umai0123 / HF_upload_sdxl_gen_img.py
Last active May 28, 2024 23:20
HF_upload_sdxl_gen_img.py
import os
import sys
import wget
import zipfile
import threading
import re
from time import sleep
import shutil
from huggingface_hub import HfApi, Repository, upload_file
import gen_img
#https://github.com/Kazuhito00/XDoG-OpenCV-Sample
import os
import cv2
import numpy as np
def DoG_filter(image, kernel_size=0, sigma=1.0, k_sigma=2.0, gamma=1.5):
g1 = cv2.GaussianBlur(image, (kernel_size, kernel_size), sigma)
g2 = cv2.GaussianBlur(image, (kernel_size, kernel_size), sigma * k_sigma)
return g1 - gamma * g2