Update app.py

app.py

@@ -15,7 +15,7 @@ from apscheduler.schedulers.background import BackgroundScheduler
 from flask import Flask, request, jsonify, Response, stream_with_context
 
 os.environ['TZ'] = 'Asia/Shanghai'
-# time.tzset()
+time.tzset()
 
 logging.basicConfig(level=logging.INFO,
                     format='%(asctime)s - %(levelname)s - %(message)s')
@@ -841,7 +841,6 @@ def handsome_images_generations():
     response_data = {}
     
     if "stable-diffusion" in model_name or model_name in ["black-forest-labs/FLUX.1-schnell", "Pro/black-forest-labs/FLUX.1-schnell","black-forest-labs/FLUX.1-dev", "black-forest-labs/FLUX.1-pro"]:
-        
         siliconflow_data = {
             "model": model_name,
             "prompt": data.get("prompt"),
@@ -876,7 +875,7 @@ def handsome_images_generations():
             if siliconflow_data["interval"] < 1 or siliconflow_data["interval"] > 4 :
                  siliconflow_data["interval"] = 2
         else:    
-            siliconflow_data["image_size"] = data.get("image_size", "1024x1024") # Use 'image_size' directly
+            siliconflow_data["image_size"] = data.get("image_size", "1024x1024") 
             siliconflow_data["prompt_enhancement"] = data.get("prompt_enhancement", False)
             seed = data.get("seed")
             if isinstance(seed, int) and 0 < seed < 9999999999:
@@ -961,7 +960,6 @@ def handsome_images_generations():
                     "data": []
                 }
 
-
             logging.info(
                 f"使用的key: {api_key}, "
                 f"总共用时: {total_time:.4f}秒, "
@@ -1016,7 +1014,6 @@ def handsome_chat_completions():
     }
     
     if model_name in image_models:
-        # Handle image generation
         user_content = ""
         messages = data.get("messages", [])
         for message in messages: