Update app.py

app.py

@@ -4,6 +4,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 from huggingface_hub import login
 import matplotlib.pyplot as plt
 import numpy as np
+import os
 
 # Login to Hugging Face with token
 login(token=os.environ["HF_TOKEN"])