memorease-api / app.py
memorease's picture
Update app.py
a6b4a26 verified
raw
history blame
526 Bytes
from fastapi import FastAPI
from pydantic import BaseModel
from transformers import pipeline
app = FastAPI()
# 🔁 MODELİ SADECE CPU'DA ÇALIŞACAK ŞEKİLDE ZORLA
generator = pipeline("text2text-generation", model="memorease/memorease-quizgen", device=-1)
class Memory(BaseModel):
description: str
@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"]}