Update app.py

app.py

@@ -37,7 +37,7 @@ model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1",
 model.to(device)
 
 # Define the image generation function
-@spaces.GPU(duration=60)
+@spaces.GPU(duration=60 * 2)
 def generate_image(prompt, enhance_prompt, negative_prompt, num_inference_steps, height, width, guidance_scale, seed, num_images_per_prompt):
     if seed == 0:
         seed = random.randint(1, 2**32-1)
@@ -55,8 +55,9 @@ def generate_image(prompt, enhance_prompt, negative_prompt, num_inference_steps,
         do_sample=True,
         temperature=0.7,
         top_p=1,
-        top_k=50,
-    )
+        top_k=50
+        )
+        prompt = tokenizer.decode(outputs[0], skip_special_tokens=True)
         
     generator = torch.Generator().manual_seed(seed)