Hugging Face's logo

Spaces:
memorease
/
MemorySpace
Runtime error

App Files Community
memorease commited on
Commit
736e2a4
·
verified ·
1 Parent(s): faf1ac3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,11 +2,12 @@ from fastapi import FastAPI
2
  from pydantic import BaseModel
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
  import torch
 
5
 
6
  app = FastAPI()
7
 
8
  # Hugging Face modelini yükle (CPU'da çalışacak şekilde)
9
- model = AutoModelForCausalLM.from_pretrained("memorease/memorease-quizgen")
10
  tokenizer = AutoTokenizer.from_pretrained("memorease/memorease-quizgen")
11
 
12
  class Memory(BaseModel):
 
2
  from pydantic import BaseModel
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
  import torch
5
+ from transformers import AutoModel
6
 
7
  app = FastAPI()
8
 
9
  # Hugging Face modelini yükle (CPU'da çalışacak şekilde)
10
+ model = AutoModel.from_pretrained("memorease/memorease-quizgen")
11
  tokenizer = AutoTokenizer.from_pretrained("memorease/memorease-quizgen")
12
 
13
  class Memory(BaseModel):