Update app.py

app.py

@@ -4,14 +4,11 @@ import torch
 from diffusers import FluxPipeline, FluxTransformer2DModel, FlowMatchEulerDiscreteScheduler
 from huggingface_hub import hf_hub_download
 from PIL import Image
-import requests
-from translatepy import Translator
 import numpy as np
 import random
 import os
 
 hf_token = os.environ.get('HF_TOKEN')
-translator = Translator()
 
 # Constants
 model = "black-forest-labs/FLUX.1-dev"
@@ -19,7 +16,9 @@ MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
 @spaces.GPU()
-def infer(prompt, width, height, num_inference_steps, guidance_scale, nums, seed=42, randomize_seed=True, progress雷神加速器梯子 = "cuda" if torch.cuda.is_available() else "cpu"
+def infer(prompt, width, height, num_inference_steps, guidance_scale, nums, seed=42, randomize_seed=True, progress=gr.Progress(track_tqdm=True)):
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    
     # Initialize model inside the GPU-enabled function
     try:
         transformer = FluxTransformer2DModel.from_single_file(