Alvin2707 commited on
Commit
80c39fd
·
verified ·
1 Parent(s): af3d431

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,8 +6,8 @@ import torch
6
  app = FastAPI()
7
 
8
  # Load model and tokenizer tanpa mendefinisikan cache_dir
9
- tokenizer = AutoTokenizer.from_pretrained("unsloth/Llama-3.2-1B-Instruct")
10
- model = AutoModelForCausalLM.from_pretrained("unsloth/Llama-3.2-1B-Instruct").to("cpu")
11
 
12
  # Definisikan model data untuk body JSON
13
  class GenerateRequest(BaseModel):
 
6
  app = FastAPI()
7
 
8
  # Load model and tokenizer tanpa mendefinisikan cache_dir
9
+ tokenizer = AutoTokenizer.from_pretrained("cahya/gpt2-small-indonesian-522M")
10
+ model = AutoModelForCausalLM.from_pretrained("cahya/gpt2-small-indonesian-522M").to("cpu")
11
 
12
  # Definisikan model data untuk body JSON
13
  class GenerateRequest(BaseModel):