Update app.py
Browse files
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-
|
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 |
|