from fastapi import FastAPI | |
# Use a pipeline as a high-level helper | |
from transformers import pipeline | |
pipe = pipeline("text-generation", model="deepseek-ai/DeepSeek-V2.5-1210", trust_remote_code=True) | |
app = FastAPI() | |
def home(): | |
return {"hello": "Detail"} | |
def ask(prompt: str): | |
result = pipe(prompt) | |
return result[0] |