Update app.py

app.py

@@ -85,9 +85,12 @@ def run_gpt4(req1, req2, prompt_type, api_key):
     return completion.choices[0].message.content.strip()
 
 def run_deepseek(req1, req2, prompt_type):
+    print("Start run deepseek")
     tokenizer = app.state.deepseek_tokenizer
     model = app.state.deepseek_model
+    print("Start prompt building")
     prompt = build_prompt(req1, req2, prompt_type)
+    print("The prompt is ", prompt)
     inputs = tokenizer([prompt], return_tensors="pt").to(model.device)
     outputs = model.generate(inputs.input_ids, max_new_tokens=256)
     return tokenizer.decode(outputs[0], skip_special_tokens=True)
@@ -122,7 +125,7 @@ def predict(request: ConflictDetectionRequest):
 
         elif request.model_choice == "DeepSeek-Reasoner":
             answer = run_deepseek(request.Req1, request.Req2, request.prompt_type)
-
+            print("Deepseek answer is : ", answer)
         elif request.model_choice == "LLaMA-3.1-8B-Instruct":
             if not hasattr(app.state, "llama_model"):
                 return JSONResponse({"error": "LLaMA not loaded (missing HF_TOKEN)"}, status_code=400)