app.py

from fastapi import FastAPI
from pydantic import BaseModel
from transformers import pipeline

# FastAPI uygulaması başlat
app = FastAPI()

# Modeli sadece CPU'da çalışacak şekilde yükle
generator = pipeline("text2text-generation", model="memorease/memorease-quizgen", device=-1)

# API'ye gönderilecek istek yapısı
class Memory(BaseModel):
    description: str

# POST endpoint: /generate
@app.post("/generate")
def generate(memory: Memory):
    prompt = f"Soru üret: {memory.description}"
    result = generator(prompt, max_length=64, do_sample=False)
    return {"question": result[0]["generated_text"]}