Create app.py

app.py

@@ -0,0 +1,28 @@
+# Install Hugging Face transformers and PyTorch
+pip install transformers torch
+ 
+# Optional: Install datasets for data loading
+pip install datasets
+
+# Load model directly
+from transformers import AutoTokenizer, AutoModelForTokenClassification
+import torch
+
+tokenizer = AutoTokenizer.from_pretrained("yonigo/distilbert-base-cased-pii-en")
+model = AutoModelForTokenClassification.from_pretrained("yonigo/distilbert-base-cased-pii-en")
+
+model.eval()
+
+text = "Hello"
+
+inputs = tokenizer(text, return_tensors="pt")
+
+with torch.no_grad():
+    outputs = model(**inputs)
+ 
+# Get predicted logits
+logits = outputs.logits
+ 
+# Get prediction
+predicted_class = torch.argmax(logits, dim=-1).item()
+print(f"Predicted class: {predicted_class}")