app/main.py

from fastapi import FastAPI
from transformers import AutoTokenizer, AutoModel

app = FastAPI()

# بارگیری مدل و توکنیزر
tokenizer = AutoTokenizer.from_pretrained("deepseek-ai/DeepSeek-R1")
model = AutoModel.from_pretrained("deepseek-ai/DeepSeek-R1")

@app.get("/greet")
def greet(name: str = "کاربر"):
    return {"message": f"سلام، {name}!"}

@app.post("/predict")
def predict(text: str):
    inputs = tokenizer(text, return_tensors="pt")
    outputs = model(**inputs)
    # پردازش خروجی‌ها و برگرداندن نتیجه
    return {"prediction": outputs}