Update classifier_model.py

Author: hasnainnqv

app/classifier_model.py

@@ -97,13 +97,11 @@ def classify_text(text, model_ai_hum, model_llm, tokenizer):
     else:
         predicted_label = "AI"
 
-        # with torch.no_grad():
-        #     outputs_llm = model_llm(input_ids, attention_mask=attention_mask)
-        #     logits_llm = outputs_llm.logits
-        #     predicted_llm_pred = np.argmax(logits_llm.cpu().numpy(), axis=1).item()
+        with torch.no_grad():
+            outputs_llm = model_llm(input_ids, attention_mask=attention_mask)
+            logits_llm = outputs_llm.logits
+            predicted_llm_pred = np.argmax(logits_llm.cpu().numpy(), axis=1).item()
 
-
-        predicted_llm_pred = 0
         if predicted_llm_pred == 0:
             predicted_label_llm = "LLAMA"
         else: