Spaces:
GIZ
/
Running on CPU Upgrade

prashant commited on
Commit
3f0df44
1 Parent(s): 7de7bf4
Files changed (1) hide show
  1. udfPreprocess/sdg_classifier.py +1 -1
udfPreprocess/sdg_classifier.py CHANGED
@@ -58,7 +58,7 @@ def sdg_classification(haystackdoc:List[Document])->Tuple[DataFrame,Series]:
58
  labels_= [(l.meta['classification']['label'],
59
  l.meta['classification']['score'],l.content,) for l in results]
60
 
61
- df = DataFrame(labels_, columns=["text","SDG","Relevancy"])
62
 
63
  # df['text'] = paraList
64
  df = df.sort_values(by="Relevancy", ascending=False).reset_index(drop=True)
 
58
  labels_= [(l.meta['classification']['label'],
59
  l.meta['classification']['score'],l.content,) for l in results]
60
 
61
+ df = DataFrame(labels_, columns=["SDG","Relevancy","text"])
62
 
63
  # df['text'] = paraList
64
  df = df.sort_values(by="Relevancy", ascending=False).reset_index(drop=True)