app ready

app.py

@@ -1,9 +1,32 @@
 import gradio as gr
+import json
+import torch
+import onnxruntime as rt
+from transformers import AutoTokenizer
 
 
-def greet(name):
-    return "Hello " + name + "!!"
+tokenizer = AutoTokenizer.from_pretrained("distilroberta-base")
 
+with open("genre_types_encoded_kaggle_onnx.json", "r") as file:
+    encode_genre_types = json.load(file)
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+
+genres = list(encode_genre_types.keys())
+
+
+inf_session = rt.InferenceSession("game-classifier-quantized-kaggle.onnx")
+input_name = inf_session.get_inputs()[0].name
+output_name = inf_session.get_outputs()[0].name
+
+
+def classify_game_genre(summary):
+    input_ids = tokenizer(summary)["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)))
+
+
+labels = gr.outputs.Label(num_top_classes=5)
+iface = gr.Interface(fn=classify_game_genre, inputs="text", outputs=labels)
+iface.launch(inline=False)

game-classifier-quantized-kaggle.onnx

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

genre_types_encoded_kaggle_onnx.json

@@ -0,0 +1 @@
+{"Casual": 0, "Indie": 1, "Sports": 2, "Action": 3, "Adventure": 4, "Strategy": 5, "Massively Multiplayer": 6, "RPG": 7, "Free to Play": 8, "Simulation": 9, "Early Access": 10, "Racing": 11, "Utilities": 12, "Education": 13, "Violent": 14, "Gore": 15, "Design & Illustration": 16, "Animation & Modeling": 17, "Game Development": 18, "Software Training": 19, "Video Production": 20}

requirements.txt

@@ -1 +1,4 @@
 gradio==3.41.2
+torch==2.0.1
+onnxruntime==1.15.1
+transformers==4.32.1