Update app.py

app.py

@@ -36,8 +36,8 @@ def predict(title, summary, tokenizer, model):
 @st.cache(suppress_st_warning=True) 
 def get_results(prediction, prediction_probs):
     ans = "Topic:\t\tConfidence:\n"
-    for (class, prob) in zip(prediction, prediction_probs):
-        ans += class + "\t\t" + str(prob) + "\n"
+    for (topic, prob) in zip(prediction, prediction_probs):
+        ans += topic + "\t\t" + str(prob) + "\n"
     return ans
 
 st.title("Arxiv articles classification")