reverse_sd

reverse_sd.py

# -*- coding: utf-8 -*-
"""Reverse SD.ipynb

Automatically generated by Colaboratory.

Original file is located at
    https://colab.research.google.com/drive/1ci11cstH7uM9SPb6q2hb7f-IiZHYvcDq
"""

!pip install clip-interrogator==0.6.0

from PIL import Image
from clip_interrogator import Config, Interrogator

image = Image.open("monalisa.png").convert('RGB')

image

ci = Interrogator(Config(clip_model_name="ViT-L-14/openai"))

print(ci.interrogate(image))

import locale
locale.getpreferredencoding = lambda: "UTF-8"

!pip install gradio

import gradio as gr 
import re

def reverse_sd(image):
    image = Image.fromarray(image)
    ci = Interrogator(Config(clip_model_name="ViT-L-14/openai"))
    result = ci.interrogate(image)
    return result

demo = gr.Interface(fn=reverse_sd, inputs="image", outputs="text")

demo.launch(share=True, debug=True)