Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
-from transformers import LlamaTokenizer, LlamaForCausalLM
+from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
-import os  # os 모듈 임포트
+import os
 
 model_id = 'Bllossom/llama-3.2-Korean-Bllossom-3B'
 
@@ -9,11 +9,11 @@ model_id = 'Bllossom/llama-3.2-Korean-Bllossom-3B'
 hf_access_token = os.getenv('HUGGINGFACEHUB_API_TOKEN')
 
 # 토크나이저와 모델 로드
-tokenizer = LlamaTokenizer.from_pretrained(
+tokenizer = AutoTokenizer.from_pretrained(
     model_id,
     use_auth_token=hf_access_token
 )
-model = LlamaForCausalLM.from_pretrained(
+model = AutoModelForCausalLM.from_pretrained(
     model_id,
     torch_dtype=torch.bfloat16,
     device_map="auto",