Upload 5 files

Question-classifier-quantized.onnx

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

README.md

@@ -1,13 +1,13 @@
 ---
-title: StackOverflow Question Classifier
-emoji: 🌍
-colorFrom: red
+title: Stackoverflow Question Classifier
+emoji: 🌖
+colorFrom: indigo
 colorTo: purple
 sdk: gradio
-sdk_version: 4.14.0
+sdk_version: 3.17.0
 app_file: app.py
 pinned: false
-license: mit
+license: apache-2.0
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

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("/Users/sadihossain/Desktop/Question_Classifier/tags_types_encoded.json", "r") as fp:
+  encode_tags_types = json.load(fp)
+
+Tags = list(encode_tags_types.keys())
+
+inf_session = rt.InferenceSession('/Users/sadihossain/Desktop/Question_Classifier/Question-classifier-quantized.onnx')
+input_name = inf_session.get_inputs()[0].name
+output_name = inf_session.get_outputs()[0].name
+
+def classify_Question_tags(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(Tags, 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)
+					

requirements.txt

@@ -0,0 +1,4 @@
+gradio==3.17.0
+onnxruntime==1.13.1
+torch==1.13.1
+transformers==4.26.0

tags_types_encoded.json

@@ -0,0 +1 @@
+{"java": 0, "angular": 1, "python": 2, "python-3.x": 3, "javascript": 4, "html": 5, "css": 6, "react-native": 7, "c#": 8, "ios": 9, "r": 10, "android": 11, "django": 12, "azure": 13, "reactjs": 14, "flutter": 15, "kotlin": 16, "laravel": 17, "c++": 18, "postgresql": 19, "c": 20, "sql": 21, "mysql": 22, "docker": 23, "swift": 24, "visual-studio-code": 25, "excel": 26, "typescript": 27, "pandas": 28, "node.js": 29, "spring-boot": 30, "next.js": 31, "php": 32, ".net": 33, "amazon-web-services": 34, "dart": 35}