update app

app.py

@@ -1,9 +1,36 @@
 import gradio as gr
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import torch
 
+DEVICE = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
+MODEL_ID = "LinguaCustodia/ClassiFin"
+TOKENIZER_ID = "Snowflake/snowflake-arctic-embed-m"
+MODEL = AutoModelForSequenceClassification.from_pretrained(MODEL_ID).to(DEVICE)
+TOKENIZER = AutoTokenizer.from_pretrained(TOKENIZER_ID)
 
-def greet(name):
-    return "Hello " + name + "!!"
 
+def classify(text):
+    inputs = TOKENIZER(text, return_tensors="pt")
+    for k, v in inputs.items():
+        inputs[k] = v.to(DEVICE)
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+    outputs = MODEL(**inputs)
+    logits = outputs.logits.cpu().clamp(min=0.0, max=1.0).squeeze().item()
+    return {
+        "finance": logits,
+        "not finance": 1.0 - logits,
+    }
+
+
+EXAMPLES = [
+    'A call option, often simply labeled a "call", is a contract between the buyer and the seller of the call option to exchange a security at a set price.',
+    "Natural language processing (NLP) is a subfield of computer science and especially artificial intelligence. It is primarily concerned with providing computers with the ability to process data encoded in natural language and is thus closely related to information retrieval, knowledge representation and computational linguistics, a subfield of linguistics.",
+]
+
+demo = gr.Interface(
+    fn=classify,
+    inputs=gr.Textbox(lines=3),
+    examples=EXAMPLES,
+    outputs=gr.Label(),
+)
 demo.launch()

requirements.txt

@@ -0,0 +1,75 @@
+aiofiles==23.2.1
+annotated-types==0.7.0
+anyio==4.8.0
+certifi==2025.1.31
+charset-normalizer==3.4.1
+click==8.1.8
+fastapi==0.115.11
+ffmpy==0.5.0
+filelock==3.17.0
+fsspec==2025.3.0
+gradio==5.20.1
+gradio-client==1.7.2
+groovy==0.1.2
+h11==0.14.0
+httpcore==1.0.7
+httpx==0.28.1
+huggingface-hub==0.29.3
+idna==3.10
+jinja2==3.1.6
+markdown-it-py==3.0.0
+markupsafe==2.1.5
+mdurl==0.1.2
+mpmath==1.3.0
+networkx==3.4.2
+numpy==2.2.3
+nvidia-cublas-cu12==12.4.5.8
+nvidia-cuda-cupti-cu12==12.4.127
+nvidia-cuda-nvrtc-cu12==12.4.127
+nvidia-cuda-runtime-cu12==12.4.127
+nvidia-cudnn-cu12==9.1.0.70
+nvidia-cufft-cu12==11.2.1.3
+nvidia-curand-cu12==10.3.5.147
+nvidia-cusolver-cu12==11.6.1.9
+nvidia-cusparse-cu12==12.3.1.170
+nvidia-cusparselt-cu12==0.6.2
+nvidia-nccl-cu12==2.21.5
+nvidia-nvjitlink-cu12==12.4.127
+nvidia-nvtx-cu12==12.4.127
+orjson==3.10.15
+packaging==24.2
+pandas==2.2.3
+pillow==11.1.0
+pydantic==2.10.6
+pydantic-core==2.27.2
+pydub==0.25.1
+pygments==2.19.1
+python-dateutil==2.9.0.post0
+python-multipart==0.0.20
+pytz==2025.1
+pyyaml==6.0.2
+regex==2024.11.6
+requests==2.32.3
+rich==13.9.4
+ruff==0.9.10
+safehttpx==0.1.6
+safetensors==0.5.3
+semantic-version==2.10.0
+setuptools==76.0.0
+shellingham==1.5.4
+six==1.17.0
+sniffio==1.3.1
+starlette==0.46.1
+sympy==1.13.1
+tokenizers==0.21.1
+tomlkit==0.13.2
+torch==2.6.0
+tqdm==4.67.1
+transformers==4.49.0
+triton==3.2.0
+typer==0.15.2
+typing-extensions==4.12.2
+tzdata==2025.1
+urllib3==2.3.0
+uvicorn==0.34.0
+websockets==15.0.1