Update app.py

app.py

@@ -1,6 +1,13 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 import torch
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
+from diffusers import StableDiffusionPipeline
+from huggingface_hub import login
+import os
+from PIL import Image
+import io
+
+login(token=os.environ["HF_TOKEN"])
 
 model_id = "mistralai/Mistral-7B-Instruct-v0.1"  # Requiere acceso
 tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=True)