Update app.py

app.py

@@ -15,7 +15,7 @@ def get_prompts(prompt_text):
 	    return text_gen(prompt_text + " realistic, 8k, cyberpunk, highly detailed, ultra super realism, realism, high graphics")
     else:
         return text_gen("")
-proc1=gr.Interface.load("models/Yntec/photoMovieX")
+proc1=gr.Interface.load("models/AchyuthGamer/ImMagician")
 
 def restart_script_periodically():
     while True:
@@ -75,14 +75,6 @@ def send_it2(inputs, noise_level, proc1=proc1):
     output2 = proc1(prompt_with_noise)
     return output2
 
-def send_itX(inputs, noise_level, proc1=proc1):
-    prompt_with_noise = add_random_noise(inputs, noise_level)
-    while queue.qsize() >= queue_threshold:
-        time.sleep(2)
-    queue.put(prompt_with_noise)
-    outputX = proc1(prompt_with_noise)
-    return outputX
-
 def send_it3(inputs, noise_level, proc1=proc1):
     prompt_with_noise = add_random_noise(inputs, noise_level)
     while queue.qsize() >= queue_threshold:
@@ -99,23 +91,6 @@ def send_it4(inputs, noise_level, proc1=proc1):
     output4 = proc1(prompt_with_noise)
     return output4
 
-def send_it5(inputs, noise_level, proc1=proc1):
-    prompt_with_noise = add_random_noise(inputs, noise_level)
-    while queue.qsize() >= queue_threshold:
-        time.sleep(2)
-    queue.put(prompt_with_noise)
-    output5 = proc1(prompt_with_noise)
-    return output5
-
-#def send_it7(inputs, noise_level, proc1=proc1):
-    #prompt_with_noise = add_random_noise(inputs, noise_level)
-    #while queue.qsize() >= queue_threshold:
-    #    time.sleep(2)
-    #queue.put(prompt_with_noise)
-    #output5 = proc1(prompt_with_noise)
-    #return output0
-
-
 with gr.Blocks(css='style.css') as demo:
     gr.HTML(
         """
@@ -190,22 +165,11 @@ with gr.Blocks(css='style.css') as demo:
             with gr.Row():
                 output3=gr.Image(label="ImMagician", show_label=False, min_width=640, object_fit="contain", height="auto")
                 output4=gr.Image(label="ImMagician", show_label=False, min_width=640, object_fit="contain", height="auto")
-        with gr.Row():
-            with gr.Row():
-                output5=gr.Image(label="ImMagician", show_label=False, min_width=640, object_fit="contain", height="auto")
-                outputX=gr.Image(label="ImMagician", show_label=False, min_width=640, object_fit="contain", height="auto")
-        #with gr.Row():
-            #with gr.Row():
-                #output0=gr.Image(label="PhotoMovieX",show_label=False,min_width=640)
         see_prompts.click(get_prompts, inputs=[input_text], outputs=[prompt], queue=False)
         run.click(send_it1, inputs=[prompt, noise_level], outputs=[output1])
-        #run.click(send_it7, inputs=[prompt, noise_level], outputs=[output0])
         run.click(send_it2, inputs=[prompt, noise_level], outputs=[output2])
         run.click(send_it3, inputs=[prompt, noise_level], outputs=[output3])
         run.click(send_it4, inputs=[prompt, noise_level], outputs=[output4])
-        run.click(send_it5, inputs=[prompt, noise_level], outputs=[output5])
-        run.click(send_itX, inputs=[prompt, noise_level], outputs=[outputX])
-        
 
         with gr.Row():
                 gr.HTML(