CrypticMonkey3 commited on
Commit
3be3d58
·
verified ·
1 Parent(s): 48b2207

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import gradio as gr
6
  from spaces import GPU # Hugging Face ZeroGPU decorator
7
 
8
  # Load model and processor
9
- model_name = "prithivMLmods/AI-vs-Deepfake-vs-Real"
10
  processor = AutoImageProcessor.from_pretrained(model_name)
11
  model = AutoModelForImageClassification.from_pretrained(model_name)
12
  model.eval()
 
6
  from spaces import GPU # Hugging Face ZeroGPU decorator
7
 
8
  # Load model and processor
9
+ model_name = "Smogy/SMOGY-Ai-images-detector"
10
  processor = AutoImageProcessor.from_pretrained(model_name)
11
  model = AutoModelForImageClassification.from_pretrained(model_name)
12
  model.eval()