king007 commited on
Commit
2abe7fa
1 Parent(s): 0aa8e95

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -8,6 +8,7 @@ from transformers import (
8
  BartForConditionalGeneration
9
  )
10
  import pandas as pd
 
11
 
12
  # model_tapex = "microsoft/tapex-large-finetuned-wtq"
13
  # tokenizer_tapex = AutoTokenizer.from_pretrained(model_tapex)
@@ -34,8 +35,9 @@ pipe_tapas = pipeline(task="table-question-answering", model="google/tapas-large
34
 
35
 
36
 
37
- def process2(query, csv_data):
38
  # csv_data={"Actors": ["Brad Pitt", "Leonardo Di Caprio", "George Clooney"], "Number of movies": ["87", "53", "69"]}
 
39
  table = pd.DataFrame.from_dict(csv_data)
40
  #microsoft
41
  encoding = tokenizer(table=table, query=query, return_tensors="pt")
 
8
  BartForConditionalGeneration
9
  )
10
  import pandas as pd
11
+ import json
12
 
13
  # model_tapex = "microsoft/tapex-large-finetuned-wtq"
14
  # tokenizer_tapex = AutoTokenizer.from_pretrained(model_tapex)
 
35
 
36
 
37
 
38
+ def process2(query, csv_dataStr):
39
  # csv_data={"Actors": ["Brad Pitt", "Leonardo Di Caprio", "George Clooney"], "Number of movies": ["87", "53", "69"]}
40
+ csv_data = json.loads(csv_dataStr)
41
  table = pd.DataFrame.from_dict(csv_data)
42
  #microsoft
43
  encoding = tokenizer(table=table, query=query, return_tensors="pt")