Super-squash branch 'main' using huggingface_hub

Co-authored-by: ooferdoodles <[email protected]>

.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,13 @@
+---
+title: Text2tags Demo
+emoji: 👀
+colorFrom: red
+colorTo: indigo
+sdk: gradio
+sdk_version: 4.36.1
+app_file: app.py
+pinned: false
+license: openrail
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,35 @@
+import gradio as gr
+from text2tags import TaggerLlama
+
+model = TaggerLlama()
+
+def predict(caption, max_tokens=128, temperature=0.8, top_k=40, top_p=0.95, repeat_penalty=1.1):
+    tags = model.predict_tags(caption, max_tokens=max_tokens, temperature=temperature,
+                              top_k=top_k, top_p=top_p, repeat_penalty=repeat_penalty)
+    return ', '.join(tags)
+
+description = """
+### Enter a caption to extract danbooru tags from it.
+This space is duplicated from [ooferdoodles/text2tags-demo](https://huggingface.co/spaces/ooferdoodles/text2tags-demo)
+"""
+
+demo = gr.Interface(
+    fn=predict,
+    inputs=[
+        gr.Textbox(label="Caption"),
+        gr.Slider(0, 256, step=16, value=128, label='max_tokens'),
+        gr.Slider(0, 2, step=0.1, value=0.8, label='temperature'),
+        gr.Slider(0, 100, step=5, value=40, label='top_k'),
+        gr.Slider(0, 2, step=0.05, value=0.95, label='top_p'),
+        gr.Slider(0, 5, step=0.1, value=1.1, label='repeat_penalty'),
+    ],
+    outputs="text",
+    title="Text2Tags",
+    description=description,
+    examples=[
+        ["Minato Aqua from hololive with pink and blue twintails in a blue maid outfit"],
+    ],
+    allow_flagging="never"
+)
+
+demo.launch()

requirements.txt

@@ -0,0 +1 @@
+text2tags-lib