removed decode

app.py

@@ -15,7 +15,6 @@ ort_model = ORTModelForFeatureExtraction.from_pretrained("hsmashiana/optimized_m
 tokenizer = AutoTokenizer.from_pretrained("hsmashiana/optimized_model_hpml")
 model3 = OnnxSetFitModel(ort_model, tokenizer, model1.model_head)
 
-decode = {0: "World", 1: "Sports", 2: "Business", 3: "Sci/Tech"}
 
 def plot_throughput_bar_chart(throughput_model1, throughput_model2):
     labels = ['Base model', 'Optimized model']
@@ -76,13 +75,13 @@ def compare_models(text):
 
     result1 = {
         "Base Model": {
-            "answer": decode[out1.numpy()[0]],
+            "answer": out1.numpy()[0],
             "average time (ms)": avg_latency_ms_model_1,
             "throughput (tokens/sec)": throughput_tokens_per_sec1
         }}
     result2 = {
         "Optimized Model": {
-            "answer": decode[out3.numpy()[0]],
+            "answer": out3.numpy()[0],
             "average time (ms)": avg_latency_ms_model_3,
             "throughput (tokens/sec)": throughput_tokens_per_sec2
         }}