memorease commited on
Commit
ad4a894
·
verified ·
1 Parent(s): a6b4a26

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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
- # 🔁 MODELİ SADECE CPU'DA ÇALIŞACAK ŞEKİLDE ZORLA
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}"