Update Demo.py

Demo.py

@@ -70,9 +70,8 @@ def fit_data(pipeline, json_data, question):
     spark_df = spark.createDataFrame([[json_data, question]]).toDF("table_json", "questions")
     model = pipeline.fit(spark_df)
     lightPipelineModel = LightPipeline(model)
-    data = f""" table_json:{json_data} questions:{question} """
-    data
-    result = lightPipelineModel.fullAnnotate(data)
+    table, data = f""" table_json:{json_data} """ , """ questions:{question} """
+    result = lightPipelineModel.fullAnnotate([table, data])
     st.write(result)
     return result