Update app.py

app.py

@@ -21,6 +21,9 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype=torch.float16).to("cpu")  # Use "cuda" if available
 
 
+class ChatRequest(BaseModel):
+    message: str
+
 def generate_sql_query(user_input: str) -> str:
     """
     Generate an SQL query from a natural language query using the DeepSeek model.