Update app.py

app.py

@@ -4,19 +4,34 @@ import pandas as pd
 
 
 
-def coding(model, text, codetext):
+def coding_v1(model, text, codetext):
     classifier = pipeline("zero-shot-classification", model=model)
     codelist = codetext.split(';')
     output = classifier(text, codelist, multi_label=True)
     return output
 
+def coding(model, text, codetext):
+    classifier = pipeline("zero-shot-classification", model=model)
+    codelist = codetext.split(';')
+    output = classifier(text, codelist, multi_label=True)
+    # keys = output.labels
+    # values = output.scores
+    keys = output['labels']
+    values = output['scores']
+    my_dict = {k: v for k, v in zip(keys, values)}
+    return my_dict
+    
 def upload_code_list(file):
     df = pd.read_excel(file.name, sheet_name='code')
     joined_data = ';'.join(df['label'].astype(str))
     return joined_data
 
 
-demo = gr.Blocks()
+css = """
+h2.svelte-1pq4gst{display:none}
+"""
+
+demo = gr.Blocks(css=css)
 
 with demo:
     gr.Markdown(
@@ -59,8 +74,10 @@ with demo:
                 variant='primary'
             )
         with gr.Column():
-            result_text = gr.JSON()
-    run_btn.click(coding, [select_model, comment_text, codelist_text], result_text, scroll_to_output=True)
+            #result_text = gr.JSON()
+            result_label = gr.Label()
+            
+    run_btn.click(coding, [select_model, comment_text, codelist_text], result_label, scroll_to_output=True)
 
 if __name__ == "__main__":
     demo.launch()