Upload 2 files

README.md

@@ -1,10 +1,9 @@
 ---
-title: img
+title: png
 emoji: 👩‍🎨
 colorFrom: red
 colorTo: yellow
 sdk: docker
 pinned: false
-app_port: 7861
 ---
 

main.py

@@ -44,7 +44,7 @@ def generate_image(image, inference_steps, guidance_scale, image_size, prompt, m
         # 如果选择了图生图模型但没有提供图像，可以返回一个错误消息或使用默认图像
         return Image.new('RGB', (512, 512), color = 'white')  # 创建一个白色的默认图像
 
-    url = f"https://api.siliconflow.cn/v1/stabilityai/{model_choice}"
+    url = f"https://api.siliconflow.cn/v1/{model_choice}"
     
     with requests.post(url, data=json.dumps(data), headers=headers) as response:
         res = response.json()
@@ -70,8 +70,20 @@ if __name__ == "__main__":
         with gr.Row():
             prompt_input = gr.Textbox(label="输入prompt", lines=3, placeholder="在此输入你的prompt...")
             with gr.Column():
-                model_choice = gr.Dropdown(['stable-diffusion-xl-base-1.0/image-to-image', 'stable-diffusion-2-1/text-to-image'], 
-                                           label='选择模型', value='stable-diffusion-xl-base-1.0/image-to-image')
+                model_choice = gr.Dropdown([
+                                           'stabilityai/stable-diffusion-3-medium/text-to-image',
+                                           'stabilityai/stable-diffusion-xl-base-1.0/text-to-image',
+                                           'stabilityai/stable-diffusion-2-1/text-to-image', 
+                                           'stabilityai/sd-turbo/text-to-image', 
+                                           'stabilityai/sdxl-turbo/text-to-image', 
+                                           'ByteDance/SDXL-Lightning/text-to-image', 
+                                           'stabilityai/stable-diffusion-xl-base-1.0/image-to-image', 
+                                           'stabilityai/stable-diffusion-2-1/image-to-image', 
+                                           'stabilityai/SDXL-Lightning/image-to-image', 
+                                           'InstantX/InstantID/image-to-image', 
+                                           'TencentARC/PhotoMaker/image-to-image'
+], 
+                                           label='选择模型', value='stabilityai/stable-diffusion-xl-base-1.0/image-to-image')
                 with gr.Row():
                     clear_btn = gr.Button("清空")
                     generate_btn = gr.Button("生成图片")