bump

requirements.txt

@@ -1,7 +1,7 @@
 fastapi[standard]==0.111.0
 Pillow==10.3.0
-transformers==4.40.1
-torch==2.3.0
+transformers==4.41.2
+torch==2.3.1
 python-dotenv==1.0.1
 filetype==1.2.0
 opencv-python-headless==4.9.0.80

src/shared/shared.py

@@ -17,13 +17,18 @@
 
 
 def check_model(model_name):
-    _model_name = model_name or default_model_name
+    try:
+        _model_name = model_name or default_model_name
 
-    classifier = pipeline(
-        "image-classification",
-        model=_model_name,
-        token=access_token,
-        device=device,
-    )
+        classifier = pipeline(
+            "image-classification",
+            model=_model_name,
+            token=access_token,
+            device=device,
+        )
 
-    return classifier
+        return classifier
+
+    except Exception as e:
+        print(e)
+        return {"error": str(e)}