teaevo commited on
Commit
4f6e66f
·
1 Parent(s): 7c29a25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import TapexTokenizer, BartForConditionalGeneration
3
  import pandas as pd
4
 
5
- model_name = "microsoft/tapex-large-sql-execution" # 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
 
@@ -29,7 +29,7 @@ iface = gr.Interface(
29
  outputs=gr.Textbox(),
30
  live=True,
31
  capture_session=True,
32
- title="Chatbot",
33
  description="Type your message in the box above, and the chatbot will respond.",
34
  )
35
 
 
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
 
 
29
  outputs=gr.Textbox(),
30
  live=True,
31
  capture_session=True,
32
+ title="ST SQL Chatbot",
33
  description="Type your message in the box above, and the chatbot will respond.",
34
  )
35