Spaces:

spark-nlp
/

TAPAS

Sleeping

abdullahmubeen10 commited on Aug 10, 2024

Commit

ce68d86

verified ·

1 Parent(s): 7d949b4

Update Demo.py

Files changed (1) hide show

Demo.py CHANGED Viewed

@@ -70,9 +70,6 @@ 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)
-    lia= result.select("answers_wtq.result", "answers_sqa.result").tolist()
-    lia
     pan = result.select("answers_wtq.result", "answers_sqa.result").toPandas()
     pan

     spark_df = spark.createDataFrame([[json_data, question]]).toDF("table_json", "questions")
     model = pipeline.fit(spark_df)
     result = model.transform(spark_df)
     pan = result.select("answers_wtq.result", "answers_sqa.result").toPandas()
     pan