Update app.py

app.py

@@ -3,6 +3,7 @@ import torch
 import time
 import spaces # Import the spaces library
 from transformers import AutoModelForCausalLM, AutoTokenizer
+from db import save_test_result
 
 # --- Configuration ---
 MODEL_ID = "Qwen/Qwen2.5-Math-1.5B" # Replace with actual ID if found
@@ -47,6 +48,17 @@ def generate_response(messages, max_length=512, temperature=0.7, top_p=0.9):
         num_generated_tokens = len(output_ids)
         response = tokenizer.decode(output_ids, skip_special_tokens=True)
         print("Generation complete.")
+
+        save_test_result(
+            analysis_mode='',
+            system_prompt='',
+            input_content='', 
+            model_response= response, 
+            generation_time='',
+            tokens_generated='', 
+            temperature='',
+            top_p='',
+            max_length='')
         return response.strip(), num_generated_tokens
 
     except Exception as e: