Skip to content

Commit 7d60ef8

Browse files
committed
2 parents 99962c8 + 0fb74f5 commit 7d60ef8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

app/classifier_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import transformers
2-
from transformers import BertForSequenceClassification, BertTokenizer
2+
from transformers import BertForSequenceClassification,BertTokenizer
33
import numpy as np
44
import torch
55
import re
@@ -108,4 +108,4 @@ def classify_text(text, model_ai_hum, model_llm, tokenizer):
108108
else:
109109
predicted_label_llm = "Gemini"
110110

111-
return {"type": predicted_label, "llm": predicted_label_llm}
111+
return {"type": predicted_label, "llm": predicted_label_llm}

0 commit comments

Comments
 (0)