Update app.py

app.py

@@ -16,20 +16,19 @@ def ensure_flash_attn():
         import flash_attn
         print("当前 flash-attn 已安装，版本：", flash_attn.__version__)
     except ImportError:
-        print("未安装 flash-attn，开始安装...")
-        
-
+        print("未安装 flash-attn")
+    
 ensure_flash_attn()
 from inferencer import UniPicV2Inferencer
 
 os.environ["HF_HUB_REQUEST_TIMEOUT"] = "60.0"
 model_path = snapshot_download(repo_id="Skywork/UniPic2-Metaquery-9B")
-qwen_vl_path = snapshot_download(repo_id="Qwen/Qwen2.5-VL-7B-Instruct-AWQ")
+qwen_vl_path = snapshot_download(repo_id="Qwen/Qwen2.5-VL-7B-Instruct")
 
 inferencer = UniPicV2Inferencer(
     model_path=model_path,
     qwen_vl_path=qwen_vl_path,
-    quant="int4"
+    quant="fp16"
 )
 #inferencer.pipeline = inferencer._init_pipeline()
 
@@ -69,7 +68,7 @@ def extract_assistant_reply(full_text):
     return full_text.replace("<|im_end|>", "").strip()
 
 
-def on_submit(history, user_msg, img_path, mode, infer_steps, cfg_scale, model_version='sd3.5-512', seed=3):
+def on_submit(history, user_msg, img_path, mode, infer_steps, cfg_scale, model_version='sd3.5-512', seed=3221):
     user_msg = user_msg.strip()
     updated_history = history + [(user_msg, None)]
     edit_tip = "✅ You can continue editing this image by switching the mode to 'Edit Image' and entering your instruction. "
@@ -244,7 +243,7 @@ with gr.Blocks(css=CSS) as demo:
     img_state = gr.State(value=None)
 
     with gr.Tabs():
-        with gr.Tab("Skywork UniPic2-Metaquery int4", elem_id="tab_item_4"):
+        with gr.Tab("Skywork UniPic2-Metaquery", elem_id="tab_item_4"):
             chatbot = gr.Chatbot(
                 elem_id="chatbot1",
                 show_label=False,
@@ -309,9 +308,7 @@ with gr.Blocks(css=CSS) as demo:
     )
     clear_btn.click(clear_all, outputs=[chatbot, img_state, mode_selector])
 
-demo.launch()
-
-
 
-#if __name__ == "__main__":
+if __name__ == "__main__":
+    demo.launch()
 #    demo.launch(server_name="0.0.0.0", debug=True, server_port=8004)