Spaces:

dejanseo
/

ai-detection-small

Running

dejanseo commited on Apr 27

Commit

1b73e0f

verified ·

1 Parent(s): 94a3584

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -101,7 +101,7 @@ def load_model_and_tokenizer(model_name):
     model.to(device).eval()
     return tokenizer, model, device
-MODEL_NAME = "dejanseo/ai-detection-small"
 try:
     tokenizer, model, device = load_model_and_tokenizer(MODEL_NAME)
 except Exception as e:

     model.to(device).eval()
     return tokenizer, model, device
+MODEL_NAME = "dejanseo/ai-cop"
 try:
     tokenizer, model, device = load_model_and_tokenizer(MODEL_NAME)
 except Exception as e: