Spaces:

spark-nlp
/

TAPAS

Sleeping

abdullahmubeen10 commited on Aug 10, 2024

Commit

6d5ba58

verified ·

1 Parent(s): 4f033e8

Update Demo.py

Files changed (1) hide show

Demo.py CHANGED Viewed

@@ -77,7 +77,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(data)
     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(question)
     st.write(result)
     return result