Upload requirements.txt with huggingface_hub

requirements.txt

@@ -1,21 +1,19 @@
-# Needle 3 JAX path, GPU build (ZeroGPU).
+# Needle 3 JAX path + BART-MNLI, GPU build (ZeroGPU).
 #
-# The pair MUST be matched. Unpinned `jax` resolves to a recent version whose
-# `jax.api_util` no longer exports `debug_info`, which flax's axes_scan calls,
-# giving: AttributeError: module 'jax.api_util' has no attribute 'debug_info'.
-# Verified with `pip install --dry-run`:
-#   jax[cuda12]==0.4.38 + flax==0.10.2  -> resolves clean
-#   jax[cuda12]==0.5.3  + flax==0.10.7  -> ResolutionImpossible
-#   jax[cuda12]==0.6.2  + flax==0.12.0  -> ResolutionImpossible
-# These are the pins cactus-needle's own `train` extra declares (flax>=0.10.2,
-# jax==0.4.38 for its metal extra).
+# The jax pair MUST be matched. Unpinned `jax` resolves to a recent version whose
+# `jax.api_util` no longer exports `debug_info`, which flax's axes_scan calls, and the
+# Space fails at import with AttributeError. The vendor's own extra pins flax>=0.10.2;
+# 0.10.2 and 0.10.4 both resolve cleanly against jax 0.4.38 while 0.10.6+ conflict.
 jax[cuda12]==0.4.38
+jaxlib==0.4.38
 flax==0.10.2
 optax
 sentencepiece
 safetensors
-numpy
 
-gradio>=4.44,<7
-cactus-needle==3.0.2
-huggingface_hub>=0.34.0
+# BART-MNLI backend. Note: the flax weight file for bart-large-mnli is 1.6 GB and the
+# ZeroGPU image is space-limited, so we load it from the Hub at runtime and run it on
+# CPU (device=-1) rather than bundling it. torch is required by the transformers
+# zero-shot pipeline even when running the flax weights.
+torch
+transformers