V0
Browse files
app.py
CHANGED
|
@@ -137,20 +137,9 @@ def setup_damage_detector(model_path, threshold=0.7):
|
|
| 137 |
def load_vit_deepfake_model(model_path, device):
|
| 138 |
"""Load the AutoAdapter model for deepfake detection"""
|
| 139 |
try:
|
| 140 |
-
|
| 141 |
-
try:
|
| 142 |
-
import importlib.util
|
| 143 |
-
if importlib.util.find_spec("transformers") is None:
|
| 144 |
-
print("Installing transformers...")
|
| 145 |
-
os.system("pip install transformers")
|
| 146 |
-
if importlib.util.find_spec("adapter_transformers") is None:
|
| 147 |
-
print("Installing adapter-transformers...")
|
| 148 |
-
os.system("pip install -U adapter-transformers")
|
| 149 |
-
except Exception as e:
|
| 150 |
-
print(f"Warning: Could not install adapter-transformers: {e}")
|
| 151 |
-
|
| 152 |
# Import AutoAdapterModel from adapter_transformers
|
| 153 |
-
from
|
| 154 |
|
| 155 |
# Create model from pretrained base
|
| 156 |
print(f"Loading AutoAdapter model for deepfake detection...")
|
|
|
|
| 137 |
def load_vit_deepfake_model(model_path, device):
|
| 138 |
"""Load the AutoAdapter model for deepfake detection"""
|
| 139 |
try:
|
| 140 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 141 |
# Import AutoAdapterModel from adapter_transformers
|
| 142 |
+
from adapters import AutoAdapterModel
|
| 143 |
|
| 144 |
# Create model from pretrained base
|
| 145 |
print(f"Loading AutoAdapter model for deepfake detection...")
|