Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ def generate_sql_query(user_input: str) -> str:
|
|
20 |
Generate an SQL query from a natural language query using the DeepSeek model.
|
21 |
"""
|
22 |
inputs = tokenizer(user_input, return_tensors="pt")
|
23 |
-
outputs = model.generate(**inputs, max_length=
|
24 |
sql_query = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
25 |
return sql_query
|
26 |
|
|
|
20 |
Generate an SQL query from a natural language query using the DeepSeek model.
|
21 |
"""
|
22 |
inputs = tokenizer(user_input, return_tensors="pt")
|
23 |
+
outputs = model.generate(**inputs, max_length=400)
|
24 |
sql_query = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
25 |
return sql_query
|
26 |
|