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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -1,12 +1,13 @@
1
  from fastapi import FastAPI
2
  from pydantic import BaseModel
3
- from transformers import LlamaForCausalLM, LlamaTokenizer
4
  import torch
5
 
6
  app = FastAPI()
7
 
8
- model = LlamaForCausalLM.from_pretrained("memorease/memorease-quizgen")
9
- tokenizer = LlamaTokenizer.from_pretrained("memorease/memorease-quizgen")
 
10
 
11
  class Memory(BaseModel):
12
  description: str
 
1
  from fastapi import FastAPI
2
  from pydantic import BaseModel
3
+ from transformers import AutoModelForCausalLM, AutoTokenizer
4
  import torch
5
 
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