Upload folder using huggingface_hub

README.md

@@ -21,7 +21,7 @@ sdk_version: 4.29.0
 * 人工审阅及智能投放
 
 ## 技术路线
-* 使用 python gradio lib 进行AI应用构建
+* USE python gradio as the interface for the AI application
 
 ## How to Deploy
 * gradio deploy

app.py

@@ -43,7 +43,7 @@ def send_generation_request(
 
     return response
 
-def generate_image(prompt, negative_prompt, aspect_ratio, seed, output_format):
+def generate_image(prompt, negative_prompt, aspect_ratio, seed, output_format, model):
     """Generates an image using the SD3 API."""
 
     host = os.environ["STABILITY_HOST"]
@@ -54,7 +54,7 @@ def generate_image(prompt, negative_prompt, aspect_ratio, seed, output_format):
         "aspect_ratio": aspect_ratio,
         "seed": seed,
         "output_format": output_format,
-        "model": "sd3-medium",  # You can change the model here
+        "model": model, # Use the selected model
         "mode": "text-to-image"
     }
 
@@ -87,9 +87,10 @@ interface = gr.Interface(
     inputs=[
         gr.Textbox(label="正向提示词", placeholder="This dreamlike digital art captures a vibrant, kaleidoscopic bird in a lush rainforest"),
         gr.Textbox(label="负向提示词", placeholder="Optional"),
-        gr.Dropdown(label="长宽比", choices=["21:9", "16:9", "3:2", "5:4", "1:1", "4:5", "2:3", "9:16", "9:21"], value="16:9"),
         gr.Number(label="生成算法随机种子", value=0),
-        gr.Dropdown(label="输出格式", choices=["jpeg", "png"], value="png")
+        gr.Dropdown(label="长宽比", choices=["21:9", "16:9", "3:2", "5:4", "1:1", "4:5", "2:3", "9:16", "9:21"], value="16:9"),
+        gr.Dropdown(label="输出格式", choices=["jpeg", "png"], value="png"),
+        gr.Dropdown(label="模型选择", choices=["sd3-large", "sd3-large-turbo", "sd3-medium"], value="sd3-medium")  # Add model dropdown
     ],
     outputs="image",
     title="Stable Diffusion 3 Image Generator",
@@ -97,4 +98,4 @@ interface = gr.Interface(
 )
 
 # Launch the interface
-interface.launch(share=True)
+interface.launch(share=False)