Update main.py

main.py

@@ -59,10 +59,7 @@ async def generate_image(prompt_input: PromptInput):
         time.sleep(2)
     queue.put(prompt_with_noise)
     output_dict = proc1(prompt_with_noise)
-output1 = output_dict['output'][0]
-
-
-
+    output1 = output_dict['output'][0]
     return {"image": output1}
 
 # Serve the HTML frontend