gbarone77 commited on
Commit
adc5de3
·
1 Parent(s): fe7997e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -70,8 +70,11 @@ def convert_question_to_SQL(question):
70
 
71
  device = 'cpu'
72
 
73
- tokenizer = AutoTokenizer.from_pretrained('gbarone77/t5-large-finetuned-wikisql-with-cols')
74
- model = T5ForConditionalGeneration.from_pretrained('gbarone77/t5-large-finetuned-wikisql-with-cols')
 
 
 
75
 
76
  translator = pipeline("translation", model="Helsinki-NLP/opus-mt-it-en")
77
 
 
70
 
71
  device = 'cpu'
72
 
73
+ #tokenizer = AutoTokenizer.from_pretrained('gbarone77/t5-large-finetuned-wikisql-with-cols')
74
+ #model = T5ForConditionalGeneration.from_pretrained('gbarone77/t5-large-finetuned-wikisql-with-cols')
75
+
76
+ tokenizer = AutoTokenizer.from_pretrained('model/', local_files_only=True)
77
+ model = T5ForConditionalGeneration.from_pretrained('model/', local_files_only=True)
78
 
79
  translator = pipeline("translation", model="Helsinki-NLP/opus-mt-it-en")
80