test mistral and flux 3

app.py

@@ -1,10 +1,15 @@
 import gradio as gr
 import numpy as np
 import random
+import os
 import spaces
 import torch
 from diffusers import DiffusionPipeline
 from transformers import pipeline
+from huggingface_hub import login
+
+hf_token = os.environ['hf_token']
+login(token=hf_token)
 
 
 dtype = torch.bfloat16
@@ -15,9 +20,6 @@ pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", tor
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
-
-
-
 def refine_prompt(prompt):
     gr.Info("Starting process")
     chatbot = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.3")