Update Demo.py

Demo.py

@@ -60,7 +60,7 @@ def create_pipeline(model, task, zeroShotLables=['']):
             .pretrained(model, "en") \
             .setInputCols(["sentence", "token"]) \
             .setOutputCol("ner") \
-            .setCaseSensitive(True) \
+            .setCaseSensitive(False) \
             .setMaxSentenceLength(512)
         
         ner_converter = NerConverter() \
@@ -221,7 +221,10 @@ except:
 
 # Initialize Spark and create pipeline
 spark = init_spark()
-pipeline = create_pipeline(model, task)
+if task == 'Zero-Shot Classification':
+    pipeline = create_pipeline(model, task, zeroShotLables)
+else:
+    pipeline = create_pipeline(model, task)
 output = fit_data(pipeline, text_to_analyze, task)
 
 # Display matched sentence
@@ -243,5 +246,4 @@ elif task == 'Zero-Shot Classification':
 
 elif task == 'Sequence Classification':
     st.markdown(f"Classified as : **{output[0]['class'][0].result}**")
-