Update main.py

main.py

@@ -30,8 +30,6 @@ def generate_image():
 
     return jsonify({"image_url": image_url})
 
-if __name__ == '__main__':
-    app.run(debug=True)
 
 
 
@@ -67,4 +65,7 @@ with gr.Blocks(css=css, theme='YTheme/Sketch') as demo:
     
     submit_btn.click(fn=generate_image, inputs=[prompt_input, code], outputs=image_output)
 
-demo.launch()
+demo.launch()
+
+if __name__ == '__main__':
+    app.run(debug=True)