import gradio as gr
import os

DESC = """# Classify sex positions for a video clip

WARNING! Leave now if you are less than 18 years old!

* total 10 classes are supported: ["Missionary", "Cowgirl", "Doggystyle", "Side Fucking", "Blowjob", "Titjob", "Pussy Eating", "Fingering", "Handjob", "Other"]
* "Other" means other classes such as SFW like kissing or talking or NSFW sucking tits.
* Input video should be sexy or NSFW ones, otherwise the prediction is undefined.
* Input should be .mp4 with less than 10 seconds, prediction for longer video clips might be ok but not guaranteed. Currently, gif file can not be input, open an issue if it is a problem. You can convert gif to mp4 in this [link](https://cloudconvert.com/gif-to-mp4) online or by [python](https://stackoverflow.com/questions/40726502/python-convert-gif-to-videomp4)
* The top-1 accuracy is around 90% for a validation set of ~7000 video clips.
* More classes such as footjob and higher accuracy are under development.
* A sample SFW [video](https://www.youtube.com/watch?v=giy37cf1msI) is in youtube and you can download this video by pasting the youtube url into this [website](https://en.savefrom.net/391GA/), you can experience this demo with this video. It is predicted correctly as "Other".
* I won't provide any NSFW samples, find by yourself.
* This repo is for video classification, for sex position classification for image, see [this repo](https://huggingface.co/porntech/sex-position) of mine.

I will soon be on job market and is now looking for full time or part time jobs focusing on developping AI models for sexy/NSFW videos/images, if you are interested in me or this work, feel free to contact porntech@126.com"""

with gr.Blocks() as demo:
  with gr.Row():
    gr.Markdown(DESC)
  with gr.Tab("sex position classifier for video"):
    gr.load("porntech/sex-position-classifier", src="spaces", hf_token=os.environ["PORNTECH_SECRET"])

demo.launch()