Hugging Face's logo

Spaces:
imiksi
/
deepcheck-api
Runtime error

App Files Community
imiksi commited on
Commit
b12f359
·
verified ·
1 Parent(s): 1718cfd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -7,10 +7,9 @@ from transformers import AutoModelForImageClassification, AutoFeatureExtractor
7
 
8
  app = FastAPI()
9
 
10
- # Load the Hugging Face deepfake model
11
- model_name = "yuvalkirstain/DeepFakeDetection"
12
- model = AutoModelForImageClassification.from_pretrained(model_name)
13
  extractor = AutoFeatureExtractor.from_pretrained(model_name)
 
14
 
15
  @app.post("/analyze")
16
  async def analyze(request: Request):
 
7
 
8
  app = FastAPI()
9
 
10
+ model_name = "prithivMLmods/open-deepfake-detection"
 
 
11
  extractor = AutoFeatureExtractor.from_pretrained(model_name)
12
+ model = AutoModelForImageClassification.from_pretrained(model_name)
13
 
14
  @app.post("/analyze")
15
  async def analyze(request: Request):