gauravsahu1990 commited on
Commit
a7a7bbb
·
verified ·
1 Parent(s): 9bee602

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -79,7 +79,7 @@ def generate_sql(question: str) -> str:
79
  f"{schema_description}\n"
80
  f"Question: {question}\n\nSQL:"
81
  )
82
-
83
  inputs = tokenizer(prompt, return_tensors="pt", truncation=True).to(model.device)
84
  with torch.no_grad():
85
  outputs = model.generate(**inputs, max_new_tokens=200, temperature=0.2, do_sample=False)
 
79
  f"{schema_description}\n"
80
  f"Question: {question}\n\nSQL:"
81
  )
82
+ logger.info(f"🧠 Generated prompt: {prompt}")
83
  inputs = tokenizer(prompt, return_tensors="pt", truncation=True).to(model.device)
84
  with torch.no_grad():
85
  outputs = model.generate(**inputs, max_new_tokens=200, temperature=0.2, do_sample=False)