add english and cuda

README.md

@@ -1,6 +1,6 @@
 ---
 title: Taiyi Stable Diffusion Chinese
-emoji: 🔥
+emoji: 🤯
 colorFrom: red
 colorTo: red
 sdk: gradio

app.py

@@ -3,7 +3,7 @@ from diffusers import StableDiffusionPipeline
 # from PIL import Image
 
 
-device="cpu"
+device="cuda"
 model_id = "IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-v0.1"
 
 # pipe_img2img = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, unet=unet).to(device)
@@ -26,8 +26,8 @@ gr.Interface(fn=infer, inputs=
     [
         # gr.Image(source="upload", type="filepath", label="原始图像"), 
         gr.Textbox(label = '提示词(prompt)'),
-        gr.Slider(2, 15, value = 7, label = '文本引导强度'),
-        gr.Slider(10, 30, value = 20, step = 1, label = '迭代次数'),
-        gr.Slider(256, 768, value = 512, step = 64, label = '宽度'),
-        gr.Slider(256, 768, value = 512, step = 64, label = '高度'),
+        gr.Slider(2, 15, value = 7, label = '文本引导强度(guidance scale)'),
+        gr.Slider(10, 30, value = 20, step = 1, label = '迭代次数(inference steps)'),
+        gr.Slider(256, 768, value = 512, step = 64, label = '宽度(width)'),
+        gr.Slider(256, 768, value = 512, step = 64, label = '高度(height)'),
     ],outputs='image').queue(concurrency_count=10).launch()