Add Space

.gitignore

@@ -0,0 +1 @@
+__pycache__/

.python-version

@@ -0,0 +1 @@
+3.12

README.md

@@ -7,6 +7,8 @@ sdk: gradio
 sdk_version: 6.5.1
 app_file: app.py
 pinned: false
+models:
+- CohereLabs/tiny-aya-global
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,144 @@
+import logging
+import os
+from collections.abc import Iterator
+
+import gradio as gr
+from cohere import ClientV2
+
+logger = logging.getLogger(__name__)
+
+model_id = "tiny-aya-global"
+
+# Initialize Cohere client
+api_key = os.getenv("COHERE_API_KEY")
+if not api_key:
+    raise ValueError("COHERE_API_KEY environment variable is required")
+client = ClientV2(api_key=api_key, client_name="hf-tiny-aya-global")
+
+
+def generate(
+    message: str,
+    history: list[dict],
+    system_prompt: str = "",
+    temperature: float = 0.3,
+    max_new_tokens: int = 700,
+) -> Iterator[str]:
+    """Stream a response from the Cohere API for the given message and conversation history."""
+    messages: list[dict[str, str]] = []
+    system_prompt = (system_prompt or "").strip()
+    if system_prompt:
+        messages.append({"role": "system", "content": system_prompt})
+
+    # Add conversation history
+    for item in history:
+        role = item.get("role")
+        content = item.get("content") or ""
+        if not isinstance(content, str):
+            content = str(content)
+        if role in ("assistant", "user") and content:
+            messages.append({"role": role, "content": content})
+
+    # Add current user message
+    current_text = str(message or "").strip()
+    if not current_text:
+        yield ""
+        return
+    messages.append({"role": "user", "content": current_text})
+
+    try:
+        # Call Cohere API using the correct event type and delta access
+        response = client.chat_stream(
+            model=model_id,
+            messages=messages,
+            temperature=temperature,
+            max_tokens=max_new_tokens,
+        )
+
+        output = ""
+        for event in response:
+            if getattr(event, "type", None) == "content-delta":
+                # event.delta.message.content.text is the streamed text
+                text = getattr(event.delta.message.content, "text", "")
+                output += text
+                yield output
+
+    except Exception:
+        logger.exception("Cohere API error")
+        gr.Warning("Something went wrong generating a response. Please try again.")
+        yield ""
+
+
+examples = [
+    ["Explica en español qué significa la palabra japonesa 'ikigai' y da un ejemplo práctico."],
+    ["اكتب فقرة قصيرة تصف غروب الشمس في الصحراء"],
+    ["Kwa nini ni muhimu kujifunza lugha zaidi ya moja? Toa sababu tatu."],
+    ["一个从未见过大海的山村孩子，第一次来到海边。用三到五句话描述他的感受。"],
+    [
+        "Translate the following sentence from Basque to French: "
+        "'Hizkuntza-eredu handiek milioika testu erabiltzen dituzte ikasteko, "
+        "baina hizkuntza txikientzat datu gutxiago dago.'"
+    ],
+    [
+        "ਜੇਕਰ ਕੋਈ ਵਿਅਕਤੀ ਪਹਿਲੀ ਵਾਰ ਵਿਦੇਸ਼ ਜਾ ਰਿਹਾ ਹੈ, ਤਾਂ ਉਸ ਨੂੰ ਕਿਹੜੀਆਂ ਗੱਲਾਂ ਦਾ "
+        "ਧਿਆਨ ਰੱਖਣਾ ਚਾਹੀਦਾ ਹੈ? ਪੰਜ ਸੁਝਾਅ ਦਿਓ।"
+    ],
+    [
+        "Eglurwch mewn tair brawddeg pam mae bioamrywiaeth yn bwysig i ecosystemau."
+    ],
+    [
+        "ถ้าคุณต้องการเริ่มต้นออกกำลังกายเป็นประจำ ควรเริ่มต้นอย่างไร? ให้คำแนะนำสามข้อ"
+    ]
+]
+
+example_labels = [
+    "Spanish — Explain 'ikigai'",
+    "Arabic — Describe a desert sunset",
+    "Swahili — Why learn multiple languages?",
+    "Chinese — A child sees the ocean",
+    "English — Basque to French translation",
+    "Punjabi — Travel tips for first-timers",
+    "Welsh — Why biodiversity is important to ecosystems",
+    "Thai — How to start exercising regularly",
+]
+
+DESCRIPTION = (
+    "**[Tiny Aya](https://huggingface.co/CohereLabs/tiny-aya-global)** is a lightweight "
+    "multilingual language model by [Cohere Labs](https://cohere.com/research). "
+    "Try chatting in any of 70+ supported languages!"
+)
+
+demo = gr.ChatInterface(
+    fn=generate,
+    chatbot=gr.Chatbot(min_height=600),
+    textbox=gr.Textbox(
+        autofocus=True,
+        placeholder="Type a message in any language... / Escribe en cualquier idioma... / أكتب بأي لغة...",
+    ),
+    additional_inputs=[
+        gr.Textbox(
+            label="System Prompt (optional)",
+            placeholder="e.g. You are a helpful multilingual assistant. Always respond in the user's language.",
+            lines=3,
+        ),
+        gr.Slider(label="Temperature", minimum=0.0, maximum=1.0, step=0.05, value=0.3),
+        gr.Slider(label="Max New Tokens", minimum=100, maximum=2000, step=10, value=700),
+    ],
+    stop_btn=False,
+    title="Tiny Aya",
+    description=DESCRIPTION,
+    examples=examples,
+    example_labels=example_labels,
+    run_examples_on_click=True,
+    cache_examples=False,
+    delete_cache=(1800, 1800),
+    save_history=True,
+)
+
+if __name__ == "__main__":
+    demo.launch(
+        theme=gr.themes.Soft(
+            primary_hue="green",
+            font=[gr.themes.GoogleFont("Inter"), "system-ui", "sans-serif"],
+        ),
+        css_paths="style.css",
+    )

pyproject.toml

@@ -0,0 +1,51 @@
+[project]
+name = "tiny-aya-hf-space"
+version = "0.1.0"
+description = "Tiny Aya HF Space"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "cohere>=5.16.1",
+    "gradio>=6.5.1",
+]
+
+[tool.ruff]
+line-length = 119
+
+[tool.ruff.lint]
+select = ["ALL"]
+ignore = [
+    "COM812", # missing-trailing-comma
+    "D203",   # one-blank-line-before-class
+    "D213",   # multi-line-summary-second-line
+    "E501",   # line-too-long
+    "SIM117", # multiple-with-statements
+    #
+    "D100",    # undocumented-public-module
+    "D101",    # undocumented-public-class
+    "D102",    # undocumented-public-method
+    "D103",    # undocumented-public-function
+    "D104",    # undocumented-public-package
+    "D105",    # undocumented-magic-method
+    "D107",    # undocumented-public-init
+    "EM101",   # raw-string-in-exception
+    "FBT001",  # boolean-type-hint-positional-argument
+    "FBT002",  # boolean-default-value-positional-argument
+    "PD901",   # pandas-df-variable-name
+    "PGH003",  # blanket-type-ignore
+    "PLR0913", # too-many-arguments
+    "PLR0915", # too-many-statements
+    "TRY003",  # raise-vanilla-args
+]
+unfixable = [
+    "F401", # unused-import
+]
+
+[tool.ruff.lint.pydocstyle]
+convention = "google"
+
+[tool.ruff.lint.per-file-ignores]
+"*.ipynb" = ["T201", "T203"]
+
+[tool.ruff.format]
+docstring-code-format = true

requirements.txt

@@ -0,0 +1,183 @@
+# This file was autogenerated by uv via the following command:
+#    uv export --no-hashes --format requirements-txt
+aiofiles==24.1.0
+    # via gradio
+annotated-doc==0.0.4
+    # via
+    #   fastapi
+    #   typer
+annotated-types==0.7.0
+    # via pydantic
+anyio==4.12.1
+    # via
+    #   gradio
+    #   httpx
+    #   starlette
+audioop-lts==0.2.2 ; python_full_version >= '3.13'
+    # via gradio
+brotli==1.2.0
+    # via gradio
+certifi==2026.1.4
+    # via
+    #   httpcore
+    #   httpx
+    #   requests
+charset-normalizer==3.4.4
+    # via requests
+click==8.3.1
+    # via
+    #   typer
+    #   uvicorn
+cohere==5.20.5
+    # via tiny-aya-hf-space
+colorama==0.4.6 ; sys_platform == 'win32'
+    # via
+    #   click
+    #   tqdm
+fastapi==0.129.0
+    # via gradio
+fastavro==1.12.1
+    # via cohere
+ffmpy==1.0.0
+    # via gradio
+filelock==3.21.2
+    # via huggingface-hub
+fsspec==2026.2.0
+    # via
+    #   gradio-client
+    #   huggingface-hub
+gradio==6.5.1
+    # via tiny-aya-hf-space
+gradio-client==2.0.3
+    # via gradio
+groovy==0.1.2
+    # via gradio
+h11==0.16.0
+    # via
+    #   httpcore
+    #   uvicorn
+hf-xet==1.2.0 ; platform_machine == 'AMD64' or platform_machine == 'aarch64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'x86_64'
+    # via huggingface-hub
+httpcore==1.0.9
+    # via httpx
+httpx==0.28.1
+    # via
+    #   cohere
+    #   gradio
+    #   gradio-client
+    #   huggingface-hub
+    #   safehttpx
+huggingface-hub==1.4.1
+    # via
+    #   gradio
+    #   gradio-client
+    #   tokenizers
+idna==3.11
+    # via
+    #   anyio
+    #   httpx
+    #   requests
+jinja2==3.1.6
+    # via gradio
+markdown-it-py==4.0.0
+    # via rich
+markupsafe==3.0.3
+    # via
+    #   gradio
+    #   jinja2
+mdurl==0.1.2
+    # via markdown-it-py
+numpy==2.4.2
+    # via
+    #   gradio
+    #   pandas
+orjson==3.11.7
+    # via gradio
+packaging==26.0
+    # via
+    #   gradio
+    #   gradio-client
+    #   huggingface-hub
+pandas==3.0.0
+    # via gradio
+pillow==12.1.1
+    # via gradio
+pydantic==2.12.5
+    # via
+    #   cohere
+    #   fastapi
+    #   gradio
+pydantic-core==2.41.5
+    # via
+    #   cohere
+    #   pydantic
+pydub==0.25.1
+    # via gradio
+pygments==2.19.2
+    # via rich
+python-dateutil==2.9.0.post0
+    # via pandas
+python-multipart==0.0.22
+    # via gradio
+pytz==2025.2
+    # via gradio
+pyyaml==6.0.3
+    # via
+    #   gradio
+    #   huggingface-hub
+requests==2.32.5
+    # via cohere
+rich==14.3.2
+    # via typer
+safehttpx==0.1.7
+    # via gradio
+semantic-version==2.10.0
+    # via gradio
+shellingham==1.5.4
+    # via
+    #   huggingface-hub
+    #   typer
+six==1.17.0
+    # via python-dateutil
+starlette==0.52.1
+    # via
+    #   fastapi
+    #   gradio
+tokenizers==0.22.2
+    # via cohere
+tomlkit==0.13.3
+    # via gradio
+tqdm==4.67.3
+    # via huggingface-hub
+typer==0.23.1
+    # via
+    #   gradio
+    #   typer-slim
+typer-slim==0.23.1
+    # via huggingface-hub
+types-requests==2.32.4.20260107
+    # via cohere
+typing-extensions==4.15.0
+    # via
+    #   anyio
+    #   cohere
+    #   fastapi
+    #   gradio
+    #   gradio-client
+    #   huggingface-hub
+    #   pydantic
+    #   pydantic-core
+    #   starlette
+    #   typing-inspection
+typing-inspection==0.4.2
+    # via
+    #   fastapi
+    #   pydantic
+tzdata==2025.3 ; sys_platform == 'emscripten' or sys_platform == 'win32'
+    # via pandas
+urllib3==2.6.3
+    # via
+    #   requests
+    #   types-requests
+uvicorn==0.40.0
+    # via gradio

style.css

@@ -0,0 +1,10 @@
+/* Center the title */
+h1 {
+  text-align: center;
+  display: block;
+}
+
+/* Center the description below the title */
+.prose {
+  text-align: center;
+}