Omkar1872 commited on
Commit
e8cfc3e
·
verified ·
1 Parent(s): 16e3c44

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -3,11 +3,15 @@ import pandas as pd
3
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
4
  import sqlite3
5
 
6
- # Load model with slow tokenizer explicitly
 
7
  model_name = "mrm8488/t5-base-finetuned-wikiSQL"
8
- tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False, local_files_only=False) # Force slow tokenizer
 
 
9
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
10
 
 
11
  def nl_to_sql(question, file):
12
  try:
13
  df = pd.read_csv(file.name)
 
3
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
4
  import sqlite3
5
 
6
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
7
+
8
  model_name = "mrm8488/t5-base-finetuned-wikiSQL"
9
+
10
+ # Force slow tokenizer
11
+ tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
12
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
13
 
14
+
15
  def nl_to_sql(question, file):
16
  try:
17
  df = pd.read_csv(file.name)