Update app.py

app.py

@@ -2,7 +2,10 @@ import streamlit as st
 import numpy as np
 import torch
 import random
-from transformers import GPT2LMHeadModel, GPT2Tokenizer, Trainer, TrainingArguments, DataCollatorForLanguageModeling
+from transformers import (
+    GPT2LMHeadModel, GPT2Tokenizer, Trainer, TrainingArguments, DataCollatorForLanguageModeling,
+    TrainerCallback  # Import TrainerCallback here
+)
 from datasets import Dataset
 from huggingface_hub import HfApi
 import plotly.graph_objects as go
@@ -10,6 +13,7 @@ import time
 from datetime import datetime
 import threading
 
+
 # Cyberpunk and Loading Animation Styling
 def setup_cyberpunk_style():
     st.markdown("""