File size: 591 Bytes
999ff86 181c2cc 999ff86 80def82 999ff86 181c2cc 999ff86 de3e460 181c2cc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from fawkes.protection import Fawkes
import gradio as gr
import os
def predict(level, img):
# print(img)
fwks = Fawkes("extractor_2", '0', 1, mode=level)
fwks.run_protection([img], format='jpeg')
splt = img.split(".")
print(os.listdir('/tmp'))
return splt[0] + "_cloaked.jpeg"
gr.Interface(fn=predict, inputs=[gr.components.Dropdown(["low", "mid", "high"], label="Protection Level"),
gr.components.Image(type='filepath')],
outputs=gr.components.Image(type="pil")).launch(show_error=True, server_name="0.0.0.0")
|