Spaces:

spark-nlp
/

TAPAS

Sleeping

abdullahmubeen10 commited on Aug 10, 2024

Commit

dfa75b2

verified ·

1 Parent(s): 5087502

Update Demo.py

Files changed (1) hide show

Demo.py CHANGED Viewed

@@ -70,7 +70,7 @@ 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)
-    result = lightPipelineModel.fullAnnotate(question)
     st.write(result)
     return result

     spark_df = spark.createDataFrame([[json_data, question]]).toDF("table_json", "questions")
     model = pipeline.fit(spark_df)
     lightPipelineModel = LightPipeline(model)
+    result = lightPipelineModel.fullAnnotate([json_data, question])
     st.write(result)
     return result