changed app.py

app.py

@@ -1,7 +1,7 @@
 import torch
 from transformers import AutoTokenizer
 import gradio as gr
-from model import LlamaForCausalLM  # Import your custom model class
+from model import LlamaForCausalLM # Import your custom model class
 
 # Load tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained("HuggingFaceTB/cosmo2-tokenizer")