Spaces:
Sleeping
Sleeping
Upload folder using huggingface_hub
Browse files
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)
|