change to openjourney model

app.py

@@ -12,8 +12,10 @@ from diffusers import StableDiffusionPipeline
 
 READ_TOKEN = os.environ.get('HF_ACCESS_TOKEN', None)
 
-model_id = "runwayml/stable-diffusion-v1-5"
+
+# model_id = "runwayml/stable-diffusion-v1-5"
 # model_id = "CompVis/stable-diffusion-v1-4"
+model_id = "prompthero/openjourney"
 
 has_cuda = torch.cuda.is_available()
 
@@ -56,15 +58,12 @@ def generate_story(prompt):
     return story, summary, gr.update(visible=True)
 
 def on_change_event(app_state):
-#     print(f'on_change_event {app_state}')
     if app_state and app_state['running'] and app_state['img']:
         img = app_state['img']
         step = app_state['step']
-#         print(f'Updating the image:! {app_state}')
         app_state['dots'] += 1
         app_state['dots'] = app_state['dots'] % 10
         message = app_state['status_msg'] + ' *' * app_state['dots']
-#         print (f'message={message}')
         return gr.update(value=app_state['img_list'], label='intermediate steps'), gr.update(value=message)
     else:
         return gr.update(label='images list'), gr.update(value='')
@@ -87,7 +86,7 @@ with gr.Blocks() as demo:
             app_state['img_list'].append(res)
             app_state['status_msg'] = f'Generating step ({step + 1})'
         
-        prompt = prompt + ' masterpiece charcoal pencil art lord of the rings illustration'
+        prompt = prompt + ' masterpiece charcoal pencil art lord of the rings illustration mdjrny-v4 style'
         img = pipe(prompt, height=512, width=512, num_inference_steps=inference_steps, callback=callback, callback_steps=2)
         app_state['running'] = False
         app_state['img'] = None
@@ -141,4 +140,4 @@ with gr.Blocks() as demo:
 if READ_TOKEN:
     demo.queue().launch()
 else:
-    demo.queue().launch(share=True, debug=True)
+    demo.queue().launch(share=True, debug=True)