Update app.py

app.py

@@ -39,6 +39,7 @@ system_prompt = "<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n\n
 
 #Function to guess whether we use the RAG or not.
 def classification_chatrag(query):
+  print(query)
   encoding = tokenizer(query, return_tensors="pt")
   encoding = {k: v.to(classifier_model.device) for k,v in encoding.items()}
 
@@ -55,6 +56,8 @@ def classification_chatrag(query):
   # Extract the float value from the tensor
   float_value = round(probs.item()*100)
 
+  print(float_value)
+
   if float_value > 50:
     status = True
     print("We activate RAG")