memorease commited on
Commit
94b39b4
·
verified ·
1 Parent(s): dda365d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -6,8 +6,12 @@ import torch
6
  app = FastAPI()
7
 
8
  # Model ve tokenizer'ı yükle
9
- model = AutoModelForCausalLM.from_pretrained("memorease/memorease-quizgen")
10
- tokenizer = AutoTokenizer.from_pretrained("memorease/memorease-quizgen")
 
 
 
 
11
 
12
  class Memory(BaseModel):
13
  description: str
 
6
  app = FastAPI()
7
 
8
  # Model ve tokenizer'ı yükle
9
+ #model = AutoModelForCausalLM.from_pretrained("memorease/memorease-quizgen")
10
+ #tokenizer = AutoTokenizer.from_pretrained("memorease/memorease-quizgen")
11
+
12
+ # Doğrudan llama modeli yükleniyor
13
+ model = LlamaForCausalLM.from_pretrained("memorease/memorease-quizgen")
14
+ tokenizer = LlamaTokenizer.from_pretrained("memorease/memorease-quizgen")
15
 
16
  class Memory(BaseModel):
17
  description: str