Hugging Face's logo

Spaces:
gbarone77
/
txt2sql_poc_v1
Runtime error

App Files Community
gbarone77 commited on
Commit
d8079fc
·
1 Parent(s): 00b1fb0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -68,6 +68,8 @@ def convert_question_to_SQL(question):
68
  else:
69
  return original_SQL
70
 
 
 
71
  tokenizer = AutoTokenizer.from_pretrained('gbarone77/t5-large-finetuned-wikisql-with-cols')
72
  model = T5ForConditionalGeneration.from_pretrained('gbarone77/t5-large-finetuned-wikisql-with-cols')
73
 
 
68
  else:
69
  return original_SQL
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