Update app.py

app.py

@@ -40,9 +40,9 @@ 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):
   encoding = tokenizer(query, return_tensors="pt")
-  encoding = {k: v.to(model_classifier.device) for k,v in encoding.items()}
+  encoding = {k: v.to(classifier_model.device) for k,v in encoding.items()}
 
-  outputs = model_classifier(**encoding)
+  outputs = classifier_model(**encoding)
 
   logits = outputs.logits
   logits.shape