Spaces:
Configuration error
Configuration error
Update app.py
Browse files
app.py
CHANGED
@@ -2,14 +2,17 @@ from fastapi import FastAPI
|
|
2 |
from pydantic import BaseModel
|
3 |
from transformers import pipeline
|
4 |
|
|
|
5 |
app = FastAPI()
|
6 |
|
7 |
-
#
|
8 |
generator = pipeline("text2text-generation", model="memorease/memorease-quizgen", device=-1)
|
9 |
|
|
|
10 |
class Memory(BaseModel):
|
11 |
description: str
|
12 |
|
|
|
13 |
@app.post("/generate")
|
14 |
def generate(memory: Memory):
|
15 |
prompt = f"Soru üret: {memory.description}"
|
|
|
2 |
from pydantic import BaseModel
|
3 |
from transformers import pipeline
|
4 |
|
5 |
+
# FastAPI uygulaması başlat
|
6 |
app = FastAPI()
|
7 |
|
8 |
+
# Modeli sadece CPU'da çalışacak şekilde yükle
|
9 |
generator = pipeline("text2text-generation", model="memorease/memorease-quizgen", device=-1)
|
10 |
|
11 |
+
# API'ye gönderilecek istek yapısı
|
12 |
class Memory(BaseModel):
|
13 |
description: str
|
14 |
|
15 |
+
# POST endpoint: /generate
|
16 |
@app.post("/generate")
|
17 |
def generate(memory: Memory):
|
18 |
prompt = f"Soru üret: {memory.description}"
|