Update app.py

app.py

@@ -16,10 +16,10 @@ def get_prompts(prompt_text):
     else:
       return MagicPrompt('')
 logging.info('loading Dreamlike Diffusion')
-DreamDiffusion=App.Interface.load('models/SG161222/Realistic_Vision_V1.4') # Credits to SG161222
-logging.info('Loading Dreamlike PhotoReal')
-DreamPhotoReal = DreamDiffusion
-DreamPhotoReal=App.Interface.load('models/dreamlike-art/dreamlike-photoreal-2.0') # Credits to Dreamlike
+DreamDiffusion=App.Interface.load('models/dreamlike-art/dreamlike-photoreal-2.0') # Credits to SG161222
+#logging.info('Loading Dreamlike PhotoReal')
+#DreamPhotoReal = DreamDiffusion
+#DreamPhotoReal=App.Interface.load('models/dreamlike-art/dreamlike-photoreal-2.0') # Credits to Dreamlike
 
 def RestartScript():
     while True:
@@ -95,45 +95,45 @@ def SendIt4(Inputs, NoiseLevel, DreamDiffusion=DreamDiffusion):
     logging.info('Done Creating Image On Thread 4')
     return Output4
 
-def SendIt5(Inputs, NoiseLevel, DreamDiffusion=DreamDiffusion):
-    logging.info('Creating Image On Thread 5')
-    NoisedPrompt = AddNoise(Inputs, NoiseLevel)
-    while queue.qsize() >= queue_threshold:
-        time.sleep(2)
-    queue.put(NoisedPrompt)
-    Output5 = DreamPhotoReal(NoisedPrompt)
-    logging.info('Done Creating Image On Thread 5')
-    return Output5
-
-def SendIt6(Inputs, NoiseLevel, DreamDiffusion=DreamDiffusion):
-    logging.info('Creating Image On Thread 6')
-    NoisedPrompt = AddNoise(Inputs, NoiseLevel)
-    while queue.qsize() >= queue_threshold:
-        time.sleep(2)
-    queue.put(NoisedPrompt)
-    Output6 = DreamPhotoReal(NoisedPrompt)
-    logging.info('Done Creating Image On Thread 6')
-    return Output6
-
-def SendIt7(Inputs, NoiseLevel, DreamDiffusion=DreamDiffusion):
-    logging.info('Creating Image On Thread 7')
-    NoisedPrompt = AddNoise(Inputs, NoiseLevel)
-    while queue.qsize() >= queue_threshold:
-        time.sleep(2)
-    queue.put(NoisedPrompt)
-    Output7 = DreamPhotoReal(NoisedPrompt)
-    logging.info('Done Creating Image On Thread 7')
-    return Output7
-
-def SendIt8(Inputs, NoiseLevel, DreamDiffusion=DreamDiffusion):
-    logging.info('Creating Image On Thread 8')
-    NoisedPrompt = AddNoise(Inputs, NoiseLevel)
-    while queue.qsize() >= queue_threshold:
-        time.sleep(2)
-    queue.put(NoisedPrompt)
-    Output8 = DreamPhotoReal(NoisedPrompt)
-    logging.info('Done Creating Image On Thread 8')
-    return Output8
+#def SendIt5(Inputs, NoiseLevel, DreamDiffusion=DreamDiffusion):
+#    logging.info('Creating Image On Thread 5')
+#    NoisedPrompt = AddNoise(Inputs, NoiseLevel)
+#    while queue.qsize() >= queue_threshold:
+#        time.sleep(2)
+#    queue.put(NoisedPrompt)
+#    Output5 = DreamPhotoReal(NoisedPrompt)
+#    logging.info('Done Creating Image On Thread 5')
+#    return Output5
+#
+#def SendIt6(Inputs, NoiseLevel, DreamDiffusion=DreamDiffusion):
+#    logging.info('Creating Image On Thread 6')
+#    NoisedPrompt = AddNoise(Inputs, NoiseLevel)
+#    while queue.qsize() >= queue_threshold:
+#        time.sleep(2)
+#    queue.put(NoisedPrompt)
+#    Output6 = DreamPhotoReal(NoisedPrompt)
+#    logging.info('Done Creating Image On Thread 6')
+#    return Output6
+#
+#def SendIt7(Inputs, NoiseLevel, DreamDiffusion=DreamDiffusion):
+#    logging.info('Creating Image On Thread 7')
+#    NoisedPrompt = AddNoise(Inputs, NoiseLevel)
+#    while queue.qsize() >= queue_threshold:
+#        time.sleep(2)
+#    queue.put(NoisedPrompt)
+#    Output7 = DreamPhotoReal(NoisedPrompt)
+#    logging.info('Done Creating Image On Thread 7')
+#    return Output7
+#
+#def SendIt8(Inputs, NoiseLevel, DreamDiffusion=DreamDiffusion):
+#   logging.info('Creating Image On Thread 8')
+#    NoisedPrompt = AddNoise(Inputs, NoiseLevel)
+#    while queue.qsize() >= queue_threshold:
+#        time.sleep(2)
+#    queue.put(NoisedPrompt)
+#    Output8 = DreamPhotoReal(NoisedPrompt)
+#    logging.info('Done Creating Image On Thread 8')
+#    return Output8
 
 logging.info('Loading Interface')
 with App.Blocks(css='style.css') as demo:
@@ -206,10 +206,10 @@ with App.Blocks(css='style.css') as demo:
                 Output2=App.Image(label='Dreamlike Diffusion 1.0',show_label=False)
                 Output3=App.Image(label='Dreamlike Diffusion 1.0',show_label=False)
                 Output4=App.Image(label='Dreamlike Diffusion 1.0',show_label=False)
-                Output5=App.Image(label='Dreamlike PhotoReal 2.0',show_label=True)
-                Output6=App.Image(label='Dreamlike PhotoReal 2.0',show_label=False)
-                Output7=App.Image(label='Dreamlike PhotoReal 2.0',show_label=False)
-                Output8=App.Image(label='Dreamlike PhotoReal 2.0',show_label=False)
+               # Output5=App.Image(label='Dreamlike PhotoReal 2.0',show_label=True)
+               # Output6=App.Image(label='Dreamlike PhotoReal 2.0',show_label=False)
+               # Output7=App.Image(label='Dreamlike PhotoReal 2.0',show_label=False)
+               # Output8=App.Image(label='Dreamlike PhotoReal 2.0',show_label=False)
         
 
         SeePrompts.click(get_prompts, inputs=[input_text], outputs=[prompt], queue=False)
@@ -218,10 +218,10 @@ with App.Blocks(css='style.css') as demo:
         Run.click(SendIt2, inputs=[prompt, neg_prompt, NoiseLevel], outputs=[Output2])
         Run.click(SendIt3, inputs=[prompt, neg_prompt, NoiseLevel], outputs=[Output3])
         Run.click(SendIt4, inputs=[prompt, neg_prompt, NoiseLevel], outputs=[Output4])
-        Run.click(SendIt5, inputs=[prompt, neg_prompt, NoiseLevel], outputs=[Output5])
-        Run.click(SendIt6, inputs=[prompt, neg_prompt, NoiseLevel], outputs=[Output6])
-        Run.click(SendIt7, inputs=[prompt, neg_prompt, NoiseLevel], outputs=[Output7])
-        Run.click(SendIt8, inputs=[prompt, neg_prompt, NoiseLevel], outputs=[Output8])
+       # Run.click(SendIt5, inputs=[prompt, neg_prompt, NoiseLevel], outputs=[Output5])
+       # Run.click(SendIt6, inputs=[prompt, neg_prompt, NoiseLevel], outputs=[Output6])
+       # Run.click(SendIt7, inputs=[prompt, neg_prompt, NoiseLevel], outputs=[Output7])
+       # Run.click(SendIt8, inputs=[prompt, neg_prompt, NoiseLevel], outputs=[Output8])
 
 
         with App.Row():