Reinstate label translation

app.py

@@ -40,9 +40,7 @@ def classify_abstracts(abstracts:Dict[str, str],batch_size=64, progress=gr.Progr
     # for i in range(0, len(abstracts), batch_size):
     #     classification.extend(pipe(abstracts_list[i:i+batch_size]))
 
-    # for pmid, abs in zip(pmids, classification):
-    #     abs['label'] = label_lookup[abs['label']]
-    #     abs['pmid'] = pmid
+    
         
     # return classification
     results = []
@@ -80,7 +78,9 @@ def classify_abstracts(abstracts:Dict[str, str],batch_size=64, progress=gr.Progr
         except Exception as e:
             print(f"Error classifying batch starting at index {i}: {str(e)}")
             continue
-    
+    for pmid, abs in zip(pmids, results):
+        abs['label'] = label_lookup[abs['label']]
+        abs['pmid'] = pmid
     progress(1.0, desc="Classification complete!")
     return results