santh-cpu
/

ai_code_detect

Model card Files Files and versions

santh-cpu commited on May 8

Commit

e93f7f1

·

verified ·

1 Parent(s): b8a9f6a

Update model.py

Files changed (1) hide show

model.py +1 -1

model.py CHANGED Viewed

@@ -183,7 +183,7 @@ detector.load_state_dict(torch.load(weights_path, map_location=DEVICE))
 detector.eval()
 def predict(code: str, threshold: float = THRESHOLD) -> dict:
-    if len(code.strip()) < 200:
         return {"pred": "too short", "prob": None}
     with torch.no_grad():

 detector.eval()
 def predict(code: str, threshold: float = THRESHOLD) -> dict:
+    if len(code.strip()) < 120:
         return {"pred": "too short", "prob": None}
     with torch.no_grad():