Update app.py

app.py

@@ -1,4 +1,11 @@
+import os
+import subprocess
+
+os.chdir('diffusers_repo') 
+subprocess.check_call(['pip', 'install', '-e', '.'])
+
 import gradio as gr
+import spaces
 import torch
 from diffusers import CosmosTextToImagePipeline
 import random
@@ -10,6 +17,7 @@ model_id = "diffusers-internal-dev/ct2i-14B"
 pipe = CosmosTextToImagePipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16)
 pipe.to("cuda")
 
+@spaces.GPU
 def generate_image(prompt, negative_prompt, seed, randomize_seed):
     """
     Generates an image based on the provided prompt and negative prompt.