Create model.py

model.py

@@ -0,0 +1,8 @@
+# Load model directly
+import os
+from dotenv import load_dotenv
+load_dotenv()
+model_name = os.getenv('MODEL_NAME') # model name for huggingface.co in .env file
+from transformers import AutoImageProcessor, AutoModelForObjectDetection
+processor = AutoImageProcessor.from_pretrained(model_name)
+model = AutoModelForObjectDetection.from_pretrained(model_name)