Update app.py

app.py

@@ -2,6 +2,7 @@ from huggingface_hub import login
 import torch
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
+import os
 
 # Authentification avec Hugging Face
 login(token=os.getenv('API_KEY'))