Update Demo.py

Demo.py

@@ -70,7 +70,11 @@ def fit_data(pipeline, json_data, question):
     spark_df = spark.createDataFrame([[json_data, question]]).toDF("table_json", "questions")
     model = pipeline.fit(spark_df)
     result = model.transform(spark_df)
-    return str(result.select("answers_wtq.result", "answers_sqa.result"))
+    
+    lia= result.select("answers_wtq.result", "answers_sqa.result").tolist()
+    lia
+    pan = result.select("answers_wtq.result", "answers_sqa.result").toPandas()
+    pan
 
 # Sidebar content
 model = st.sidebar.selectbox(
@@ -172,7 +176,6 @@ output = fit_data(pipeline, table_json_str, text_to_analyze)
 st.markdown("---")
 st.subheader("Processed Output")
 
-output
 # # Check if output is available
 # if output:
 #     results_wtq = output[0][0] if output[0][0] else "No results found."