Update fine_tuning.py

fine_tuning.py

@@ -201,7 +201,7 @@ def generate_answer(prompt, max_tokens=200):
     if prompt.strip() == "":
         return "Please enter a prompt!"
 
-    if not validate_query(query):
+    if not validate_query(prompt):
         print("Query rejected: Not finance-related.")
         return "Query rejected: Please ask finance-related questions."