memorease commited on
Commit
f70a549
·
verified ·
1 Parent(s): ee74553

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -1,13 +1,12 @@
1
  from fastapi import FastAPI
2
  from pydantic import BaseModel
3
- from transformers import AutoTokenizer, AutoModelForCausalLM
4
  import torch
5
 
6
  app = FastAPI()
7
 
8
- # MODELİ DOĞRU ŞEKİLDE 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
 
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