Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
 import torch
-from transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import LlamaTokenizer, AutoModelForCausalLM
 
 MODEL_ID = "vietrix/viena-60m"
 
-tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, use_fast=False)
+tokenizer = LlamaTokenizer.from_pretrained(MODEL_ID, legacy=True)
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_ID,
     torch_dtype=torch.float32,