AIcoder35235 commited on
Commit
4460cb3
·
verified ·
1 Parent(s): 89f6876

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ app.add_middleware(
20
 
21
  # --- MODEL LOADING ---
22
  print("Booting Security Node. Loading Deepfake ViT model into memory...")
23
- pipe = pipeline("image-classification", model="dima806/deepfake_vs_real_image_detection")
24
  print("Agent Ready!")
25
 
26
  # --- REQUEST SCHEMA ---
 
20
 
21
  # --- MODEL LOADING ---
22
  print("Booting Security Node. Loading Deepfake ViT model into memory...")
23
+ pipe = pipeline("image-classification", model="prithivMLmods/deepfake-detector-model-v1")
24
  print("Agent Ready!")
25
 
26
  # --- REQUEST SCHEMA ---