Spaces:

istassiy
/

ysda_2022_ml2_hw3

Runtime error

istassiy commited on Mar 28, 2022

Commit

d276604

1 Parent(s): de1989c

commit from

Files changed (1) hide show

app.py CHANGED Viewed

@@ -87,5 +87,5 @@ else:
         padding=True, truncation=True, return_tensors="pt"
       )
     )
-    results = get_top_predictions(raw_predictions[0].numpy())
     st.markdown(f"{results}")

         padding=True, truncation=True, return_tensors="pt"
       )
     )
+    results = get_top_predictions(raw_predictions[0][0].numpy())
     st.markdown(f"{results}")