Spaces:
Sleeping
Sleeping
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 |
-
|
| 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
|