Hugging Face's logo

Spaces:
teaevo
/
gradio
Sleeping

App Files Community
teaevo commited on
Commit
d087072
·
1 Parent(s): 4f6e66f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,7 +2,8 @@ import gradio as gr
2
  from transformers import TapexTokenizer, BartForConditionalGeneration
3
  import pandas as pd
4
 
5
- model_name = "microsoft/tapex-large-finetuned-wikisql" # You can change this to any other model from the list above
 
6
  tokenizer = TapexTokenizer.from_pretrained(model_name)
7
  model = BartForConditionalGeneration.from_pretrained(model_name)
8
 
 
2
  from transformers import TapexTokenizer, BartForConditionalGeneration
3
  import pandas as pd
4
 
5
+ #model_name = "microsoft/tapex-large-finetuned-wikisql" # You can change this to any other model from the list above
6
+ model_name = "microsoft/tapex-large-finetuned-wtq"
7
  tokenizer = TapexTokenizer.from_pretrained(model_name)
8
  model = BartForConditionalGeneration.from_pretrained(model_name)
9