WarTitan2077 commited on
Commit
207921c
·
verified ·
1 Parent(s): fa16495

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -35
app.py DELETED
@@ -1,35 +0,0 @@
1
- !pip install fastapi uvicorn transformers torch
2
-
3
- from fastapi import FastAPI, HTTPException
4
- from pydantic import BaseModel
5
- from transformers import AutoTokenizer, AutoModelForSequenceClassification
6
- import torch
7
-
8
- app = FastAPI()
9
-
10
- # Load tokenizer and model once on startup
11
- tokenizer = AutoTokenizer.from_pretrained("WarTitan2077/Number-Classifier")
12
- model = AutoModelForSequenceClassification.from_pretrained("WarTitan2077/Number-Classifier")
13
- model.eval()
14
-
15
- class InputNumbers(BaseModel):
16
- numbers: list[str] # expects ["5", "6", "7"] or similar
17
-
18
- def predict(numbers):
19
- input_text = ", ".join(numbers)
20
- inputs = tokenizer(input_text, return_tensors="pt")
21
- with torch.no_grad():
22
- outputs = model(**inputs)
23
- logits = outputs.logits
24
- probabilities = torch.softmax(logits, dim=1).tolist()
25
- return probabilities
26
-
27
- @app.post("/predict")
28
- async def predict_numbers(data: InputNumbers):
29
- if len(data.numbers) != 3:
30
- raise HTTPException(status_code=400, detail="Exactly 3 numbers required.")
31
- preds = predict(data.numbers)
32
- return {"predictions": preds}
33
-
34
-
35
- !uvicorn app:app --reload