media-engine 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
media_engine/_version.py CHANGED
@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
28
28
  commit_id: COMMIT_ID
29
29
  __commit_id__: COMMIT_ID
30
30
 
31
- __version__ = version = '0.1.0'
32
- __version_tuple__ = version_tuple = (0, 1, 0)
31
+ __version__ = version = '0.1.1'
32
+ __version_tuple__ = version_tuple = (0, 1, 1)
33
33
 
34
34
  __commit_id__ = commit_id = None
@@ -33,22 +33,38 @@ def _run_model_checks(check_id: str) -> None:
33
33
  _model_check_status[check_id] = "running"
34
34
 
35
35
  try:
36
- # Test Qwen 2B
37
- logger.info("Testing Qwen 2B model...")
38
- start = time.time()
36
+ # Test Qwen 2B (requires qwen extras: pip install media-engine[qwen])
39
37
  try:
40
- _get_qwen_model("Qwen/Qwen2-VL-2B-Instruct")
41
- results["qwen_2b"] = {
42
- "canLoad": True,
43
- "error": None,
44
- "loadTimeSeconds": round(time.time() - start, 1),
45
- }
46
- unload_qwen_model()
38
+ import importlib.util
39
+
40
+ if importlib.util.find_spec("transformers") is None:
41
+ results["qwen_2b"] = {
42
+ "canLoad": False,
43
+ "error": "transformers not installed. Install with: pip install media-engine[qwen]",
44
+ "loadTimeSeconds": 0,
45
+ }
46
+ else:
47
+ logger.info("Testing Qwen 2B model...")
48
+ start = time.time()
49
+ try:
50
+ _get_qwen_model("Qwen/Qwen2-VL-2B-Instruct")
51
+ results["qwen_2b"] = {
52
+ "canLoad": True,
53
+ "error": None,
54
+ "loadTimeSeconds": round(time.time() - start, 1),
55
+ }
56
+ unload_qwen_model()
57
+ except Exception as e:
58
+ results["qwen_2b"] = {
59
+ "canLoad": False,
60
+ "error": str(e),
61
+ "loadTimeSeconds": round(time.time() - start, 1),
62
+ }
47
63
  except Exception as e:
48
64
  results["qwen_2b"] = {
49
65
  "canLoad": False,
50
66
  "error": str(e),
51
- "loadTimeSeconds": round(time.time() - start, 1),
67
+ "loadTimeSeconds": 0,
52
68
  }
53
69
 
54
70
  # Test Whisper large-v3
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: media-engine
3
- Version: 0.1.0
3
+ Version: 0.1.1
4
4
  Summary: AI-powered video extraction API for metadata, transcripts, faces, scenes, objects, and more
5
5
  Project-URL: Repository, https://github.com/thetrainroom/media-engine
6
6
  Project-URL: Issues, https://github.com/thetrainroom/media-engine/issues
@@ -28,6 +28,8 @@ Requires-Dist: psutil>=5.9.0
28
28
  Requires-Dist: pydantic-settings>=2.0.0
29
29
  Requires-Dist: pydantic>=2.0.0
30
30
  Requires-Dist: scenedetect[opencv]>=0.6.0
31
+ Requires-Dist: tf-keras>=2.16.0
32
+ Requires-Dist: transformers>=4.40.0
31
33
  Requires-Dist: uvicorn>=0.27.0
32
34
  Requires-Dist: webrtcvad>=2.0.10
33
35
  Provides-Extra: cpu
@@ -58,7 +60,6 @@ Provides-Extra: qwen
58
60
  Requires-Dist: accelerate>=0.30.0; extra == 'qwen'
59
61
  Requires-Dist: qwen-vl-utils>=0.0.8; extra == 'qwen'
60
62
  Requires-Dist: torch>=2.0.0; extra == 'qwen'
61
- Requires-Dist: transformers>=4.40.0; extra == 'qwen'
62
63
  Description-Content-Type: text/markdown
63
64
 
64
65
  # Media Engine
@@ -8,7 +8,7 @@ cli/scenes.py,sha256=W3DE6VTVOUo4pjbPLfX3iQIQ1LSI9Jo7n5sIZGG9dZc,1741
8
8
  cli/telemetry.py,sha256=z7UqxoGMY2L0QEkA3hlNZsP3paXREXxKkWpVVzCboJY,2029
9
9
  cli/transcript.py,sha256=CFgaxP_jFDQf-J7U-_1NK7ADx5zoLvJ-8_F2zSaCc68,2280
10
10
  media_engine/__init__.py,sha256=T0S3Jxe8MktWwlldioaPxRLRYIlp3Nw8GaIiQ6kaGrw,216
11
- media_engine/_version.py,sha256=5jwwVncvCiTnhOedfkzzxmxsggwmTBORdFL_4wq0ZeY,704
11
+ media_engine/_version.py,sha256=m8HxkqoKGw_wAJtc4ZokpJKNLXqp4zwnNhbnfDtro7w,704
12
12
  media_engine/app.py,sha256=pnSNIyqQrTXad39x5ZZ1iskkF8WE9SWwici7C7llTMM,1987
13
13
  media_engine/cli.py,sha256=V41KS4samQjsYAP0064RcQ1pHS4Zs6Q94yUBDTTNMKE,293
14
14
  media_engine/config.py,sha256=W7dUw4VioD9ZHkqX6E-iW8xxst_b_JKcCjPydfF9YIQ,20677
@@ -57,14 +57,14 @@ media_engine/extractors/metadata/tesla.py,sha256=EihXp-abCw_MpR1e1FSFCnckEz_7n1h
57
57
  media_engine/routers/__init__.py,sha256=8gf_Jo9TpSkuHg0nwQboqUxI9jioEyVCimhO-lWN-Zs,477
58
58
  media_engine/routers/batch.py,sha256=VGdAZFUeRz4X0LWu6OK-UuiW9OHNt1Pe51xPfd4j6EI,2767
59
59
  media_engine/routers/health.py,sha256=vNYSNG-B0MoM9dxzOYB5v-12-VKJkvBpXXCYCH1itZg,2748
60
- media_engine/routers/models.py,sha256=6-wxgoeJd4MLz-pHgqze_eOAdM-HNN1HXfKnnHsJ03E,6998
60
+ media_engine/routers/models.py,sha256=Y_CBvD9Q8gp3rg1_VuvEuv2lsZCsnL6aIIhB1m5pFQc,7739
61
61
  media_engine/routers/settings.py,sha256=meeBmQVE_9PhRrDMFOF30v0k2u20k2b2fxlaQB9FQ9w,3033
62
62
  media_engine/routers/utils.py,sha256=Cx6uWQldV8AC79sotVMSlRtWXTMZbSc8DWXdSBog7ys,4552
63
63
  media_engine/utils/__init__.py,sha256=UXpqgpt1NU8NTsP_iRPfiP20_OpmGqEoZS6jVmRpjXo,162
64
64
  media_engine/utils/logging.py,sha256=yOnrYMAfwqtJLOGnhiIJbzVU8m8MYF39acuIpU4A_oA,1965
65
65
  media_engine/utils/memory.py,sha256=OZB0yl21u-lj-VwmO8RrRMd_CURxeBgKZrb6QA3OYWU,1186
66
- media_engine-0.1.0.dist-info/METADATA,sha256=rScqmPz7ESlorrxpONPIo43qT9N0pECPwqZnDscif7U,8044
67
- media_engine-0.1.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
68
- media_engine-0.1.0.dist-info/entry_points.txt,sha256=vdEMWxqs_gFTni4BGrX_iCP_f2HhC6XIz8ExQvyMpcE,338
69
- media_engine-0.1.0.dist-info/licenses/LICENSE,sha256=UOl7jpYGV9IlJlBilZxQ5IoOKhDkbwUqvLDwQk2tsFw,1069
70
- media_engine-0.1.0.dist-info/RECORD,,
66
+ media_engine-0.1.1.dist-info/METADATA,sha256=w2WZuflfBUzaLXeXAt2qtrDz3FBBcYtfzUtUGUS0tLU,8059
67
+ media_engine-0.1.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
68
+ media_engine-0.1.1.dist-info/entry_points.txt,sha256=vdEMWxqs_gFTni4BGrX_iCP_f2HhC6XIz8ExQvyMpcE,338
69
+ media_engine-0.1.1.dist-info/licenses/LICENSE,sha256=UOl7jpYGV9IlJlBilZxQ5IoOKhDkbwUqvLDwQk2tsFw,1069
70
+ media_engine-0.1.1.dist-info/RECORD,,