Update app.py

app.py

@@ -12,7 +12,7 @@ import random
 
 from share_btn import community_icon_html, loading_icon_html, share_js
 
-MODELS = ["stabilityai/stable-diffusion-2-1-base", "stabilityai/stable-diffusion-2-1", "stabilityai/stable-diffusion-2"]
+MODELS = ["stabilityai/stable-diffusion-2-1", "stabilityai/stable-diffusion-2-1-base", "stabilityai/stable-diffusion-2"]
 
 #word_list_dataset = load_dataset("stabilityai/word-list", data_files="list.txt")
 #word_list = word_list_dataset["train"]['text']
@@ -33,7 +33,7 @@ def infer(prompt, negative, scale, model_id=MODELS[0], hf_token="", images=[]):
     if not hf_token: hf_token = os.getenv("HF_TOKEN", None)
     client = InferenceClient(model=model_id, token=hf_token)
     for i in range(4):
-        image = client.text_to_image(prompt=prompt, negative_prompt=negative, guidance_scale=scale, seed=gen_random_seed())
+        image = client.text_to_image(prompt=prompt, negative_prompt=negative, guidance_scale=scale, num_inference_steps=45, seed=gen_random_seed())
         if image: images.append(image)
         yield images
     #payload = {'prompt': prompt, 'negative_prompt': negative, 'guidance_scale': scale}