fix: model 3 config

.gitignore

@@ -11,4 +11,5 @@ models/*
 *.cpython-310.pyc
 */inference_logs/*
 hf_inference_logs/*.json
-hf_inference_logs/*
+hf_inference_logs/*
+.gradio/flagged/*

app_optimized.py

@@ -79,7 +79,7 @@ MODEL_PATHS = {
 CLASS_NAMES = {
     "model_1": ['artificial', 'real'],
     "model_2": ['AI Image', 'Real Image'],
-    "model_3": ['AI', 'Real'],
+    "model_3": ['artificial', 'human'],
     "model_4": ['AI', 'Real'],
     "model_5": ['Realism', 'Deepfake'],
     "model_6": ['ai_gen', 'human'],
@@ -289,7 +289,7 @@ for model_key, hf_model_path in MODEL_PATHS.items():
         if model_key == "model_5": 
             model_instance = pipeline("image-classification", model=hf_model_path, device=device)
             preprocess_func = preprocess_resize_224
-        elif model_key in ["model_3", "model_4"]:
+        elif model_key in ["model_4"]:
             current_processor = AutoFeatureExtractor.from_pretrained(hf_model_path, device=device)
             model_instance = AutoModelForImageClassification.from_pretrained(hf_model_path).to(device)