Update app.py

app.py

@@ -37,7 +37,7 @@ MAX_IMAGE_SIZE = 2048
 OUTPUT_DIR = os.getenv("OUTPUT_DIR", "./outputs")
 
 
-
+@spaces.CPU
 def generate_replicate(
     prompt: str,
     negative_prompt: str = "",
@@ -278,5 +278,4 @@ with gr.Blocks(css="style.css", theme="NoCrypt/miku@1.2.1") as demo:
         api_name="run",
     )
 
-if __name__ == "__main__":
-    demo.queue(max_size=20).launch(debug=IS_COLAB, share=IS_COLAB)
+demo.queue(max_size=20).launch(debug=IS_COLAB, share=IS_COLAB)