initial commit

.gitignore

@@ -0,0 +1,2 @@
+.venv/
+.env

README.md

@@ -8,6 +8,9 @@ sdk_version: 6.15.2
 python_version: '3.13'
 app_file: app.py
 pinned: false
+base_model: google-bert/bert-base-uncased
+datasets:
+    - sh0416/ag_news
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,47 @@
+import gradio as gr
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import torch
+
+labels = ['World', 'Sports', 'Business', 'Sci/Tech']
+
+def infer(title, description):
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
+    model_name = "mafgit/news-classifier"
+    model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=4).to(device)
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+    model.eval()
+
+    inputs = tokenizer(
+        title,
+        description,
+        max_length=128,
+        padding='max_length',
+        truncation=True,
+        return_tensors='pt'
+    )
+
+    input_ids = inputs['input_ids'].to(device)
+    attention_mask = inputs['attention_mask'].to(device)
+
+    with torch.no_grad():
+        outputs = model(
+            input_ids=input_ids,
+            attention_mask=attention_mask
+        )
+
+    max_idx = outputs.logits.argmax(dim=1).item()
+    output = labels[max_idx]
+    return output
+
+
+ui = gr.Interface(
+    fn=infer,
+    inputs=["text", "text"],
+    outputs=["text"],
+    title="News Classifier",
+    description="Classify news articles into World, Sports, Business, or Sci/Tech categories."
+)
+
+ui.launch()

requirements.txt

@@ -0,0 +1,59 @@
+annotated-doc==0.0.4
+annotated-types==0.7.0
+anyio==4.13.0
+brotli==1.2.0
+certifi==2026.5.20
+click==8.4.1
+colorama==0.4.6
+fastapi==0.136.3
+filelock==3.29.0
+fsspec==2026.4.0
+gradio==6.15.2
+gradio_client==2.5.0
+groovy==0.1.2
+h11==0.16.0
+hf-gradio==0.4.1
+hf-xet==1.5.0
+httpcore==1.0.9
+httpx==0.28.1
+huggingface_hub==1.17.0
+idna==3.17
+Jinja2==3.1.6
+markdown-it-py==4.2.0
+MarkupSafe==3.0.3
+mdurl==0.1.2
+mpmath==1.3.0
+networkx==3.6.1
+numpy==2.4.6
+orjson==3.11.9
+packaging==26.2
+pandas==3.0.3
+pillow==12.2.0
+pydantic==2.13.4
+pydantic_core==2.46.4
+pydub==0.25.1
+Pygments==2.20.0
+python-dateutil==2.9.0.post0
+python-multipart==0.0.30
+pytz==2026.2
+PyYAML==6.0.3
+regex==2026.5.9
+rich==15.0.0
+safehttpx==0.1.7
+safetensors==0.7.0
+semantic-version==2.10.0
+setuptools==81.0.0
+shellingham==1.5.4
+six==1.17.0
+starlette==1.2.1
+sympy==1.14.0
+tokenizers==0.22.2
+tomlkit==0.14.0
+torch==2.12.0
+tqdm==4.67.3
+transformers==5.9.0
+typer==0.25.1
+typing-inspection==0.4.2
+typing_extensions==4.15.0
+tzdata==2026.2
+uvicorn==0.48.0