Spaces:

hsmashiana
/

HPML_proj

Runtime error

hsmashiana commited on May 6

Commit

6ea93fd

•

1 Parent(s): 74bcc75

updated decoder

Files changed (1) hide show

app.py CHANGED Viewed

@@ -81,7 +81,7 @@ def compare_models(text):
         }}
     result2 = {
         "Optimized Model": {
-            "answer": decode[out3[0]],
             "average time (ms)": avg_latency_ms_model_3,
             "throughput (tokens/sec)": throughput_tokens_per_sec2
         }}

         }}
     result2 = {
         "Optimized Model": {
+            "answer": decode[out3.numpy()[0]],
             "average time (ms)": avg_latency_ms_model_3,
             "throughput (tokens/sec)": throughput_tokens_per_sec2
         }}