Spaces:
Running
on
Zero
Running
on
Zero
update: fix some bug
Browse files
app.py
CHANGED
|
@@ -175,10 +175,10 @@ def greet(mode, language, input_text):
|
|
| 175 |
sub_texts = segment_text(split_texts, language=language)
|
| 176 |
detected = []
|
| 177 |
if language == "Chinese":
|
| 178 |
-
model = DiscrepancyEstimator(
|
| 179 |
prob_estimator = zh_prob_estimator
|
| 180 |
else:
|
| 181 |
-
model = DiscrepancyEstimator(
|
| 182 |
prob_estimator = en_prob_estimator
|
| 183 |
model.eval()
|
| 184 |
for i, sub_text in enumerate(sub_texts):
|
|
|
|
| 175 |
sub_texts = segment_text(split_texts, language=language)
|
| 176 |
detected = []
|
| 177 |
if language == "Chinese":
|
| 178 |
+
model = DiscrepancyEstimator(pretrained_ckpt="JiachenFu/Qwen2-0.5B-detectanyllm-detector-zh").to(device)
|
| 179 |
prob_estimator = zh_prob_estimator
|
| 180 |
else:
|
| 181 |
+
model = DiscrepancyEstimator(pretrained_ckpt="JiachenFu/Qwen2-0.5B-detectanyllm-detector-en").to(device)
|
| 182 |
prob_estimator = en_prob_estimator
|
| 183 |
model.eval()
|
| 184 |
for i, sub_text in enumerate(sub_texts):
|