Update app.py

app.py

@@ -6,7 +6,6 @@ import os
 import subprocess
 import sys
 import atexit
-import time
 
 API_KEY = os.environ.get("SILICONFLOW_API_KEY")
 if not API_KEY:
@@ -20,26 +19,13 @@ HEADERS = {
     "Authorization": f"Bearer {API_KEY}"
 }
 
-# Cloudflared 安装和设置
-def install_cloudflared():
-    if sys.platform.startswith('linux'):
-        subprocess.run(["curl", "-L", "--output", "cloudflared", "https://github.com/cloudflare/cloudflared/releases/latest/download/cloudflared-linux-amd64"])
-        subprocess.run(["chmod", "+x", "cloudflared"])
-    elif sys.platform == 'darwin':
-        subprocess.run(["brew", "install", "cloudflare/cloudflare/cloudflared"])
-    elif sys.platform == 'win32':
-        subprocess.run(["powershell", "-Command", "Invoke-WebRequest -Uri https://github.com/cloudflare/cloudflared/releases/latest/download/cloudflared-windows-amd64.exe -OutFile cloudflared.exe"])
-    else:
-        raise OSError("Unsupported operating system")
+CLOUDFLARED_TOKEN = "eyJhIjoiYmRlYmEzN2E5MzRmOTYwMjM4MjZjMzhjZGU4N2U1NDUiLCJ0IjoiNDI4Mjg3NWEtZmE3Yi00OTNmLTkzMmEtYmUxZDQ3MDUyZDIyIiwicyI6Ik9UVmhNRGt4TldJdE9EazJZaTAwWlRGakxXSXdaREl0WldVeE5UVTBZbVprTUdFdyJ9"
 
 cloudflared_process = None
 
 def start_cloudflared(port):
     global cloudflared_process
-    if sys.platform.startswith('linux') or sys.platform == 'darwin':
-        cloudflared_process = subprocess.Popen(["./cloudflared", "tunnel", "--url", f"http://localhost:{port}"])
-    elif sys.platform == 'win32':
-        cloudflared_process = subprocess.Popen(["cloudflared.exe", "tunnel", "--url", f"http://localhost:{port}"])
+    cloudflared_process = subprocess.Popen(["cloudflared", "tunnel", "--url", f"http://localhost:{port}", "run", "--token", CLOUDFLARED_TOKEN])
     atexit.register(stop_cloudflared)
 
 def stop_cloudflared():
@@ -48,7 +34,6 @@ def stop_cloudflared():
         cloudflared_process.terminate()
         cloudflared_process.wait()
 
-# 其余函数保持不变
 def generate_image(model, prompt, image_size, batch_size=1, num_inference_steps=20):
     data = {
         "model": model,
@@ -83,7 +68,7 @@ def generate_image(model, prompt, image_size, batch_size=1, num_inference_steps=
 
 def use_gemma_model(prompt, task):
     data = {
-        "model": "google/gemma-2-9b-it",
+        "model": "Qwen/Qwen2.5-7B-Instruct",
         "messages": [
             {"role": "system", "content": f"You are an AI assistant that helps with {task}. Respond concisely."},
             {"role": "user", "content": prompt}
@@ -159,8 +144,6 @@ with gr.Blocks() as demo:
 
 if __name__ == "__main__":
     port = 7860  # 或者您想使用的其他端口
-    print("Installing Cloudflared...")
-    install_cloudflared()
     print("Starting Cloudflared...")
     start_cloudflared(port)
     print("Cloudflared started. Launching Gradio app...")