Pujan-Dev commited on
Commit
50c40fa
·
1 Parent(s): df1b2ad

fixed the download issue

Browse files
config.py CHANGED
@@ -28,7 +28,7 @@ class Config:
28
  REAL_FORGED_MODEL_LOCAL_PATH = os.getenv("REAL_FORGED_MODEL_LOCAL_PATH", "Model/real_forged/fft_cnn_model_78.pth")
29
  DOCUMENT_FORGERY_MODEL_REPO_ID = os.getenv(
30
  "DOCUMENT_FORGERY_MODEL_REPO_ID",
31
- IMAGE_CLASSIFIER_REPO_ID
32
  )
33
  DOCUMENT_FORGERY_MODEL_FILENAME = os.getenv(
34
  "DOCUMENT_FORGERY_MODEL_FILENAME",
 
28
  REAL_FORGED_MODEL_LOCAL_PATH = os.getenv("REAL_FORGED_MODEL_LOCAL_PATH", "Model/real_forged/fft_cnn_model_78.pth")
29
  DOCUMENT_FORGERY_MODEL_REPO_ID = os.getenv(
30
  "DOCUMENT_FORGERY_MODEL_REPO_ID",
31
+ REPO_ID_LANG
32
  )
33
  DOCUMENT_FORGERY_MODEL_FILENAME = os.getenv(
34
  "DOCUMENT_FORGERY_MODEL_FILENAME",
features/real_forged_classifier/controller.py CHANGED
@@ -67,6 +67,7 @@ class documentForger:
67
  def is_forged(self, document_file: IO) -> dict:
68
  # Ensure a document model is loaded
69
  if not hasattr(models, 'doc_model') or models.doc_model is None:
 
70
  return {"detail": "Document forgery model not available."}
71
 
72
  # Read file bytes
 
67
  def is_forged(self, document_file: IO) -> dict:
68
  # Ensure a document model is loaded
69
  if not hasattr(models, 'doc_model') or models.doc_model is None:
70
+ _downloadmodel = Config.DOCUMENT_FORGERY_MODEL_PATH
71
  return {"detail": "Document forgery model not available."}
72
 
73
  # Read file bytes