Initial commit

README.md

@@ -1,13 +1,15 @@
 ---
 title: Multilabel Scifi Tags Classifier
-emoji: 🌍
-colorFrom: red
-colorTo: indigo
+emoji: 🏢
+colorFrom: green
+colorTo: purple
 sdk: gradio
-sdk_version: 4.38.1
+sdk_version: 3.50.0
 app_file: app.py
 pinned: false
 license: mit
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+HuggingFace Spaces App URL: https://huggingface.co/spaces/zzarif/Multilabel-Scifi-Tags-Classifier

app.py

@@ -0,0 +1,27 @@
+import gradio as gr
+import onnxruntime as rt
+from transformers import AutoTokenizer
+import torch, json
+
+tokenizer = AutoTokenizer.from_pretrained("distilroberta-base")
+
+with open("tag_types_encoded.json", "r") as fp:
+  encode_tag_types = json.load(fp)
+
+genres = list(encode_tag_types.keys()) 
+
+inf_session = rt.InferenceSession('question-classifier-quantized.onnx')
+input_name = inf_session.get_inputs()[0].name
+output_name = inf_session.get_outputs()[0].name
+
+def classify_question_tags(description):
+  input_ids = tokenizer(description)['input_ids'][:512]
+  logits = inf_session.run([output_name], {input_name: [input_ids]})[0]
+  logits = torch.FloatTensor(logits)
+  probs = torch.sigmoid(logits)[0]
+  return dict(zip(genres, map(float, probs))) 
+
+label = gr.outputs.Label(num_top_classes=5)
+iface = gr.Interface(fn=classify_question_tags, inputs="text", outputs=label)
+iface.launch(inline=False)
+					

question-classifier-quantized.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f0b77757323507e5e90e1ff6edebc2ef87aec8520883a9c6fa5e918e8ed75eb1
+size 82681062

requirements.txt

@@ -0,0 +1,4 @@
+gradio==3.50.0
+onnxruntime==1.18.0
+torch==1.13.1
+transformers==4.41.2

tag_types_encoded.json

@@ -0,0 +1,162 @@
+{
+  "the-lord-of-the-rings": 0,
+  "tolkiens-legendarium": 1,
+  "the-one-ring": 2,
+  "star-wars": 3,
+  "the-force": 4,
+  "the-empire-strikes-back": 5,
+  "harry-potter": 6,
+  "technology": 7,
+  "story-identification": 8,
+  "short-stories": 9,
+  "time-travel": 10,
+  "suggested-order": 11,
+  "back-to-the-future": 12,
+  "behind-the-scenes": 13,
+  "j-r-r-tolkien": 14,
+  "dc": 15,
+  "character-identification": 16,
+  "hogwarts": 17,
+  "a-new-hope": 18,
+  "the-hitchhikers-guide-to-the-galaxy": 19,
+  "batman": 20,
+  "the-matrix": 21,
+  "gandalf": 22,
+  "isaac-asimov": 23,
+  "horcrux": 24,
+  "magical-theory": 25,
+  "star-trek": 26,
+  "star-trek-tng": 27,
+  "the-force-awakens": 28,
+  "uss-enterprise": 29,
+  "star-wars-legends": 30,
+  "spaceship": 31,
+  "marvel": 32,
+  "marvel-cinematic-universe": 33,
+  "sauron": 34,
+  "religion": 35,
+  "comics": 36,
+  "sith": 37,
+  "darth-vader": 38,
+  "return-of-the-jedi": 39,
+  "revenge-of-the-sith": 40,
+  "albus-dumbledore": 41,
+  "severus-snape": 42,
+  "luke-skywalker": 43,
+  "obi-wan-kenobi": 44,
+  "star-trek-tos": 45,
+  "voldemort": 46,
+  "the-return-of-the-king": 47,
+  "discworld": 48,
+  "game-of-thrones": 49,
+  "a-song-of-ice-and-fire": 50,
+  "jk-rowling": 51,
+  "rogue-one": 52,
+  "death-star": 53,
+  "the-hobbit": 54,
+  "powers": 55,
+  "darth-sidious": 56,
+  "borg": 57,
+  "adaptation-comparison": 58,
+  "the-fellowship-of-the-ring": 59,
+  "history-of": 60,
+  "iron-man": 61,
+  "weapon": 62,
+  "elves": 63,
+  "artificial-intelligence": 64,
+  "magic": 65,
+  "spells": 66,
+  "books": 67,
+  "aliens": 68,
+  "science-fiction-genre": 69,
+  "the-hunger-games": 70,
+  "avengers": 71,
+  "thor-marvel": 72,
+  "the-silmarillion": 73,
+  "dragons": 74,
+  "muggles": 75,
+  "the-last-jedi": 76,
+  "droids": 77,
+  "star-trek-voyager": 78,
+  "names": 79,
+  "languages": 80,
+  "canon": 81,
+  "jedi": 82,
+  "vampire": 83,
+  "middle-earth": 84,
+  "captain-america": 85,
+  "the-avengers-2012": 86,
+  "kylo-ren": 87,
+  "doctor-who": 88,
+  "star-trek-data": 89,
+  "the-rise-of-skywalker": 90,
+  "x-men": 91,
+  "babylon-5": 92,
+  "the-phantom-menace": 93,
+  "hard-sci-fi": 94,
+  "battlestar-galactica": 95,
+  "video-games": 96,
+  "movie": 97,
+  "star-trek-2009": 98,
+  "futurama": 99,
+  "avengers-age-of-ultron": 100,
+  "the-incredible-hulk": 101,
+  "character-motivation": 102,
+  "lightsaber": 103,
+  "star-trek-ds9": 104,
+  "episode-identification": 105,
+  "horror": 106,
+  "magical-creatures": 107,
+  "space-exploration": 108,
+  "computers": 109,
+  "superman": 110,
+  "object-identification": 111,
+  "klingon": 112,
+  "nolan-batman-trilogy": 113,
+  "stargate": 114,
+  "the-terminator": 115,
+  "the-terminator-series": 116,
+  "robots": 117,
+  "death-eaters": 118,
+  "firefly": 119,
+  "alien-franchise": 120,
+  "spider-man": 121,
+  "avengers-infinity-war": 122,
+  "inspiration": 123,
+  "dc-extended-universe": 124,
+  "tv": 125,
+  "quotes": 126,
+  "wolverine": 127,
+  "attack-of-the-clones": 128,
+  "clones": 129,
+  "dune": 130,
+  "the-clone-wars": 131,
+  "the-wheel-of-time": 132,
+  "quidditch": 133,
+  "soft-sci-fi": 134,
+  "wandlore": 135,
+  "science": 136,
+  "novel": 137,
+  "robert-a-heinlein": 138,
+  "the-walking-dead": 139,
+  "avengers-endgame": 140,
+  "x-men-cinematic-universe": 141,
+  "dystopia": 142,
+  "parallel-universe": 143,
+  "star-trek-discovery": 144,
+  "avatar-the-last-airbender": 145,
+  "post-apocalyptic": 146,
+  "alien-invasion": 147,
+  "stargate-sg1": 148,
+  "star-trek-enterprise": 149,
+  "space": 150,
+  "fantastic-beasts": 151,
+  "cartoon": 152,
+  "young-adult": 153,
+  "anime": 154,
+  "space-colonization": 155,
+  "warhammer-40k": 156,
+  "telepathy": 157,
+  "childrens-novel": 158,
+  "anthology-book": 159
+}