Tameem7 commited on
Commit
ed0c76f
·
1 Parent(s): 5a27052

change model dir

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -249,7 +249,7 @@ def show_sample_predictions(num_samples: int = 10) -> str:
249
 
250
 
251
  # Determine model directory (for HF Spaces, check environment variable or use default)
252
- MODEL_DIR = os.getenv("MODEL_DIR", "prompt-injection-detector/checkpoint-5628")
253
 
254
  # Load model and data on startup
255
  print("Initializing model and dataset...")
 
249
 
250
 
251
  # Determine model directory (for HF Spaces, check environment variable or use default)
252
+ MODEL_DIR = os.getenv("MODEL_DIR", "Prompt-Classifier")
253
 
254
  # Load model and data on startup
255
  print("Initializing model and dataset...")