Deploy Gradio app loading model from HF Hub

app.py

@@ -0,0 +1,83 @@
+import os
+os.environ["CUDA_VISIBLE_DEVICES"] = "-1"  # CPU
+
+import gradio as gr
+import tensorflow as tf
+from huggingface_hub import hf_hub_download
+from transformers import AutoTokenizer, TFAutoModel
+
+
+@tf.keras.utils.register_keras_serializable()
+class DistilBertLayer(tf.keras.layers.Layer):
+    def __init__(self, model_name="vinai/bertweet-base", **kwargs):
+        super().__init__(**kwargs)
+        self.model_name = model_name
+        self.bert = TFAutoModel.from_pretrained(model_name, from_pt=True)
+
+    def call(self, inputs):
+        input_ids, attention_mask = inputs
+        outputs = self.bert(
+            input_ids=input_ids,
+            attention_mask=attention_mask,
+            training=False
+        )
+        return outputs.last_hidden_state
+
+    def get_config(self):
+        config = super().get_config()
+        config.update({"model_name": self.model_name})
+        return config
+
+
+# 1) Repo donde subiste el .keras (MODELS, no Spaces)
+MODEL_REPO = "tomy07417/disaster-tweets-bertweet-gru"  # <-- CAMBIÁ ESTO
+MODEL_FILE = "bertweet_gru_model.keras"                # <-- nombre exacto en el repo
+
+# 2) Descarga con cache (no lo baja cada vez)
+model_path = hf_hub_download(
+    repo_id=MODEL_REPO,
+    filename=MODEL_FILE,
+    repo_type="model"
+)
+
+# 3) Cargar el modelo desde el path descargado
+model = tf.keras.models.load_model(
+    model_path,
+    custom_objects={"DistilBertLayer": DistilBertLayer},
+    compile=False
+)
+
+tokenizer = AutoTokenizer.from_pretrained("vinai/bertweet-base")
+
+
+def predict(text):
+    inputs = tokenizer(
+        [text],
+        max_length=50,
+        truncation=True,
+        padding="max_length",
+        return_tensors="tf"
+    )
+
+    input_ids = inputs["input_ids"]
+    attention_mask = inputs["attention_mask"]
+
+    # si tu salida es (1,) sigmoid:
+    prob = model.predict([input_ids, attention_mask])[0][0]
+    pred = bool(prob > 0.5)
+
+    return {"prob": float(prob), "pred": pred}
+
+
+demo = gr.Interface(
+    fn=predict,
+    inputs=gr.Textbox(lines=3, label="Tweet"),
+    outputs=gr.JSON(label="Result"),
+    title="Tweet classifier",
+    description="Paste a tweet in English"
+)
+
+if __name__ == "__main__":
+    # En Spaces NO uses share=True
+    demo.launch()
+

requirements.txt

@@ -0,0 +1,199 @@
+absl-py==2.3.1
+annotated-types==0.7.0
+anyio==4.11.0
+argon2-cffi==25.1.0
+argon2-cffi-bindings==25.1.0
+arrow==1.4.0
+asttokens==3.0.0
+astunparse==1.6.3
+async-lru==2.0.5
+attrs==25.4.0
+babel==2.17.0
+beautifulsoup4==4.14.2
+bleach==6.3.0
+blis==1.3.0
+catalogue==2.0.10
+certifi==2025.10.5
+cffi==2.0.0
+charset-normalizer==3.4.4
+click==8.3.0
+cloudpathlib==0.23.0
+comm==0.2.3
+confection==0.1.5
+contourpy==1.3.3
+cycler==0.12.1
+cymem==2.0.11
+debugpy==1.8.17
+decorator==5.2.1
+defusedxml==0.7.1
+emoji==2.15.0
+en_core_web_sm @ https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.8.0/en_core_web_sm-3.8.0-py3-none-any.whl#sha256=1932429db727d4bff3deed6b34cfc05df17794f4a52eeb26cf8928f7c1a0fb85
+executing==2.2.1
+fastjsonschema==2.21.2
+filelock==3.20.0
+flatbuffers==25.9.23
+fonttools==4.60.1
+fqdn==1.5.1
+fsspec==2025.10.0
+gast==0.6.0
+gensim==4.4.0
+google-pasta==0.2.0
+grpcio==1.76.0
+gradio>=4.0
+h11==0.16.0
+h5py==3.15.1
+hf-xet==1.2.0
+httpcore==1.0.9
+httpx==0.28.1
+huggingface-hub==0.36.0
+idna==3.11
+ipykernel==7.1.0
+ipython==9.6.0
+ipython_pygments_lexers==1.1.1
+isoduration==20.11.0
+jedi==0.19.2
+Jinja2==3.1.6
+joblib==1.5.2
+json5==0.12.1
+jsonpointer==3.0.0
+jsonschema==4.25.1
+jsonschema-specifications==2025.9.1
+jupyter-events==0.12.0
+jupyter-lsp==2.3.0
+jupyter_client==8.6.3
+jupyter_core==5.9.1
+jupyter_server==2.17.0
+jupyter_server_terminals==0.5.3
+jupyterlab==4.4.10
+jupyterlab_pygments==0.3.0
+jupyterlab_server==2.28.0
+keras==3.12.0
+kiwisolver==1.4.9
+langcodes==3.5.0
+language_data==1.3.0
+lark==1.3.1
+libclang==18.1.1
+marisa-trie==1.3.1
+Markdown==3.10
+markdown-it-py==4.0.0
+MarkupSafe==3.0.3
+matplotlib==3.10.7
+matplotlib-inline==0.2.1
+mdurl==0.1.2
+mistune==3.1.4
+ml_dtypes==0.5.3
+mpmath==1.3.0
+murmurhash==1.0.13
+namex==0.1.0
+narwhals==2.10.2
+nbclient==0.10.2
+nbconvert==7.16.6
+nbformat==5.10.4
+nest-asyncio==1.6.0
+networkx==3.5
+nltk==3.9.2
+notebook==7.4.7
+notebook_shim==0.2.4
+numpy==2.3.4
+nvidia-cublas-cu12==12.8.4.1
+nvidia-cuda-cupti-cu12==12.8.90
+nvidia-cuda-nvrtc-cu12==12.8.93
+nvidia-cuda-runtime-cu12==12.8.90
+nvidia-cudnn-cu12==9.10.2.21
+nvidia-cufft-cu12==11.3.3.83
+nvidia-cufile-cu12==1.13.1.3
+nvidia-curand-cu12==10.3.9.90
+nvidia-cusolver-cu12==11.7.3.90
+nvidia-cusparse-cu12==12.5.8.93
+nvidia-cusparselt-cu12==0.7.1
+nvidia-nccl-cu12==2.27.5
+nvidia-nvjitlink-cu12==12.8.93
+nvidia-nvshmem-cu12==3.3.20
+nvidia-nvtx-cu12==12.8.90
+opt_einsum==3.4.0
+optree==0.17.0
+packaging==25.0
+pandas==2.3.3
+pandocfilters==1.5.1
+parso==0.8.5
+pexpect==4.9.0
+pillow==12.0.0
+platformdirs==4.5.0
+plotly==6.4.0
+preshed==3.0.10
+prometheus_client==0.23.1
+prompt_toolkit==3.0.52
+protobuf==6.33.0
+psutil==7.1.3
+ptyprocess==0.7.0
+pure_eval==0.2.3
+pycparser==2.23
+pydantic==2.12.3
+pydantic_core==2.41.4
+Pygments==2.19.2
+pyparsing==3.2.5
+python-dateutil==2.9.0.post0
+python-json-logger==4.0.0
+pytz==2025.2
+PyYAML==6.0.3
+pyzmq==27.1.0
+referencing==0.37.0
+regex==2025.11.3
+requests==2.32.5
+rfc3339-validator==0.1.4
+rfc3986-validator==0.1.1
+rfc3987-syntax==1.1.0
+rich==14.2.0
+rpds-py==0.28.0
+safetensors==0.6.2
+scikit-learn==1.7.2
+scipy==1.16.3
+seaborn==0.13.2
+Send2Trash==1.8.3
+sentence-transformers==5.1.2
+setuptools==80.9.0
+shellingham==1.5.4
+six==1.17.0
+smart_open==7.4.4
+sniffio==1.3.1
+soupsieve==2.8
+spacy==3.8.7
+spacy-legacy==3.0.12
+spacy-loggers==1.0.5
+srsly==2.5.1
+stack-data==0.6.3
+sympy==1.14.0
+tensorboard==2.20.0
+tensorboard-data-server==0.7.2
+tensorflow_cpu==2.20.0
+termcolor==3.2.0
+terminado==0.18.1
+tf_keras==2.20.1
+thinc==8.3.6
+threadpoolctl==3.6.0
+tinycss2==1.4.0
+tokenizers==0.22.1
+torch==2.9.1
+torchaudio==2.9.1
+torchvision==0.24.1
+tornado==6.5.2
+tqdm==4.67.1
+traitlets==5.14.3
+transformers==4.57.1
+triton==3.5.1
+typer==0.20.0
+typing-inspection==0.4.2
+typing_extensions==4.15.0
+tzdata==2025.2
+uri-template==1.3.0
+urllib3==2.5.0
+wasabi==1.1.3
+wcwidth==0.2.14
+weasel==0.4.1
+webcolors==25.10.0
+webencodings==0.5.1
+websocket-client==1.9.0
+Werkzeug==3.1.3
+wheel==0.45.1
+wrapt==2.0.0
+xgboost==3.1.1