set-up

.gitignore

@@ -0,0 +1,4 @@
+__pycache__/
+*.pyc
+harmsense/
+.env

requirements.txt

@@ -1 +1,13 @@
-huggingface_hub==0.25.2
+huggingface_hub==0.25.2
+torch
+torchvision
+gradio
+requests
+transformers
+accelerate
+Pillow
+python-dotenv
+numpy
+opencv-python
+bitsandbytes
+pydantic

src/config.py

@@ -0,0 +1,23 @@
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+class Config:
+    """Application configuration class."""
+
+    # Hugging Face Configuration
+    HF_TOKEN = os.getenv("HF_TOKEN")
+
+    # Model Configuration
+    DETR_MODEL_NAME = "facebook/detr-resnet-50"
+    QWEN_MODEL_NAME = "Qwen/Qwen2.5-Omni-7B"
+
+    @classmethod
+    def validate_config(cls):
+        """Validate the configuration values."""
+        if not cls.HF_TOKEN:
+            raise ValueError("HF_TOKEN is not set in the environment variables.")
+        
+# Validate configuration on import
+Config.validate_config()