abdullahmubeen10 commited on
Commit
fc355e4
·
verified ·
1 Parent(s): 12ca89f

Update Demo.py

Browse files
Files changed (1) hide show
  1. Demo.py +3 -3
Demo.py CHANGED
@@ -70,9 +70,9 @@ def fit_data(pipeline, json_data, question):
70
  spark_df = spark.createDataFrame([[json_data, question]]).toDF("table_json", "questions")
71
  model = pipeline.fit(spark_df)
72
  lightPipelineModel = LightPipeline(model)
73
- question
74
- json_data
75
- result = lightPipelineModel.fullAnnotate([question, json_data])
76
  st.write(result)
77
  return result
78
 
 
70
  spark_df = spark.createDataFrame([[json_data, question]]).toDF("table_json", "questions")
71
  model = pipeline.fit(spark_df)
72
  lightPipelineModel = LightPipeline(model)
73
+ data = f""" table_json:{json_data} questions:{question} """
74
+ data
75
+ result = lightPipelineModel.fullAnnotate(data)
76
  st.write(result)
77
  return result
78