Update app.py

app.py

@@ -53,15 +53,15 @@ models =[
     "digiplay/PotoPhotoRealism_v1",
 ]
 
-##bor = len(models)
-##current = random.randint(1, bor)
+bor = len(models)
+current = random.randint(1, bor)
 
-def plex(prompt,modil):
+def plex(modil,prompt,neg_prompt):
     pipe = accelerator.prepare(AutoPipelineForText2Image.from_pretrained(""+modil+"", torch_dtype=torch.float32))
     pipe = accelerator.prepare(pipe.to("cpu"))
-    image = pipe(prompt=prompt, num_inference_steps=10).images[0]
+    image = pipe(prompt=prompt, negative_prompt=neg_prompt,num_inference_steps=10).images[0]
     return image
 
-iface = gr.Interface(fn=plex,inputs=[gr.Dropdown(choices=models, type="value", value=models[0]), gr.Textbox(label="Prompt"), gr.Textbox(label="negative_prompt", value="low quality, bad quality")],outputs=gr.Image(label="Generated Output Image"), title="AutoPipelineForText2Image_SD_Multi",description="AutoPipelineForText2Image_SD_Multi")
+iface = gr.Interface(fn=plex,inputs=[gr.Dropdown(choices=models, type="value", value=models[current]), gr.Textbox(label="Prompt"), gr.Textbox(label="negative_prompt", value="low quality, bad quality")],outputs=gr.Image(label="Generated Output Image"), title="AutoPipelineForText2Image_SD_Multi",description="AutoPipelineForText2Image_SD_Multi")
 iface.queue(max_size=1)
 iface.launch(max_threads=1)