Update app.py

app.py

@@ -3,7 +3,7 @@ import spaces
 import torch
 import torch.amp as amp
 
-from transformers import MistralForCausalLM, AutoTokenizer, pipeline
+from transformers import MistralForCausalLM, LlamaTokenizer, pipeline
 
 repo_id = "appmana/Cosmos-1.0-Prompt-Upsampler-12B-Text2World-hf"
 
@@ -12,7 +12,7 @@ model = MistralForCausalLM.from_pretrained(
     attn_implementation="flash_attention_2",
     torch_dtype=torch.bfloat16
 )
-tokenizer = AutoTokenizer.from_pretrained(repo_id)
+tokenizer = LlamaTokenizer.from_pretrained(repo_id)
 pipe = pipeline(
     "text-generation",
     model=model,