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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,7 +4,8 @@ from transformers import pipeline
4
 
5
  app = FastAPI()
6
 
7
- generator = pipeline("text2text-generation", model="memorease/memorease-quizgen")
 
8
 
9
  class Memory(BaseModel):
10
  description: str
 
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