fixed model error

app.py

@@ -2,7 +2,7 @@ import torch
 from torch import nn
 from tokenizers import Tokenizer
 import gradio as gr
-from model import TransformerModel, config
+from model import LlamaForCausalLM  # Import your custom model class
 
 # Load your custom tokenizer from tokenizer.json
 tokenizer = Tokenizer.from_file("vocab.json")