Spaces:

dejanseo
/

ai-detection-small

Running

dejanseo commited on Apr 14

Commit

4f473c9

verified ·

1 Parent(s): 4db9ce2

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -59,7 +59,7 @@ def load_model_and_tokenizer(model_name):
     return tokenizer, model, torch.device(device_type)
 # Load model and tokenizer using the cached function
-MODEL_NAME = "dejanseo/ai-detection-base"
 try:
     tokenizer, model, device = load_model_and_tokenizer(MODEL_NAME)
 except Exception as e:

     return tokenizer, model, torch.device(device_type)
 # Load model and tokenizer using the cached function
+MODEL_NAME = "dejanseo/ai-detection-small"
 try:
     tokenizer, model, device = load_model_and_tokenizer(MODEL_NAME)
 except Exception as e: