astrosbd commited on
Commit
5df63f8
·
1 Parent(s): 249cb29
Files changed (1) hide show
  1. app.py +2 -13
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
- # First try to install the adapter-transformers library if not present
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 adapter_transformers import AutoAdapterModel
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...")